Alfredo

Runtime error

laverdes commited on Apr 10

Commit

1438534

1 Parent(s): b76efe0

chore: tts model=suno/bark-small

Files changed (1) hide show

Gradio_UI.py CHANGED Viewed

@@ -131,7 +131,7 @@ def pull_messages_from_step(
 text_to_speech_pipe = pipeline(
     task="text-to-speech",
-    model="facebook/fastspeech2-en-ljspeech",  #  "suno/bark-small",
     device = 0 if torch.cuda.is_available() else "cpu",
     torch_dtype=torch.float16,
     )

 text_to_speech_pipe = pipeline(
     task="text-to-speech",
+    model="suno/bark-small",  #  "suno/bark-small",
     device = 0 if torch.cuda.is_available() else "cpu",
     torch_dtype=torch.float16,
     )