Kremon96 commited on
Commit
8757087
·
verified ·
1 Parent(s): 559790e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -24,7 +24,8 @@ def generate_tts(text, temperature, repetition_penalty, speaker_selection, refer
24
  model_path="OuteAI/OuteTTS-0.3-1B",
25
  tokenizer_path="OuteAI/OuteTTS-0.3-1B",
26
  dtype=torch.bfloat16,
27
- device="cpu"
 
28
  )
29
  interface = outetts.InterfaceHF(model_version="0.3", cfg=model_config)
30
 
 
24
  model_path="OuteAI/OuteTTS-0.3-1B",
25
  tokenizer_path="OuteAI/OuteTTS-0.3-1B",
26
  dtype=torch.bfloat16,
27
+ device="cuda" if torch.cuda.is_available() else "cpu"
28
+
29
  )
30
  interface = outetts.InterfaceHF(model_version="0.3", cfg=model_config)
31