fix 2
Browse files
app.py
CHANGED
@@ -50,7 +50,7 @@ def load_f5tts():
|
|
50 |
model_type="F5TTS_Base", # Ajuste o nome do modelo se necessário
|
51 |
ckpt_file=ckpt_path,
|
52 |
vocab_file=os.path.join(os.path.dirname(ckpt_path), "vocab.txt"), # Caminho para o arquivo vocab.txt
|
53 |
-
device="cuda" if
|
54 |
use_ema=True
|
55 |
)
|
56 |
|
|
|
50 |
model_type="F5TTS_Base", # Ajuste o nome do modelo se necessário
|
51 |
ckpt_file=ckpt_path,
|
52 |
vocab_file=os.path.join(os.path.dirname(ckpt_path), "vocab.txt"), # Caminho para o arquivo vocab.txt
|
53 |
+
device="cuda" if torch.cuda.is_available() else "cpu", # Define o dispositivo
|
54 |
use_ema=True
|
55 |
)
|
56 |
|