Spaces:

OuteAI
/

OuteTTS-0.3-1B-Demo

Running on Zero

edwko commited on Jan 15

Commit

da104f5

verified ·

1 Parent(s): 18a86b4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,8 @@ def generate_tts(
     model_config = outetts.HFModelConfig_v2(
         model_path="OuteAI/OuteTTS-0.3-1B",
         tokenizer_path="OuteAI/OuteTTS-0.3-1B",
-        dtype=torch.bfloat16
     )
     interface = outetts.InterfaceHF(model_version="0.3", cfg=model_config)

     model_config = outetts.HFModelConfig_v2(
         model_path="OuteAI/OuteTTS-0.3-1B",
         tokenizer_path="OuteAI/OuteTTS-0.3-1B",
+        dtype=torch.bfloat16,
+        device="cuda"
     )
     interface = outetts.InterfaceHF(model_version="0.3", cfg=model_config)