Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
#1
by
not-lain
- opened
app.py
CHANGED
@@ -215,7 +215,7 @@ def generate(audio_path, ins, speed, alpha, beta, embedding, steps=200):
|
|
215 |
|
216 |
other_tts = tts.StyleTTS2(model_checkpoint_path='./epoch_2nd_00012.pth', config_path="models/config_ft.yml")
|
217 |
if torch.cuda.is_available():
|
218 |
-
other_tts.
|
219 |
else:
|
220 |
other_tts.device = "cpu"
|
221 |
|
|
|
215 |
|
216 |
other_tts = tts.StyleTTS2(model_checkpoint_path='./epoch_2nd_00012.pth', config_path="models/config_ft.yml")
|
217 |
if torch.cuda.is_available():
|
218 |
+
other_tts.device = "cuda"
|
219 |
else:
|
220 |
other_tts.device = "cpu"
|
221 |
|