text2speech

Runtime error

hynt commited on Jul 11

Commit

0dc060d

verified ·

1 Parent(s): fff0f9b

Update f5_tts/infer/utils_infer.py

Files changed (1) hide show

f5_tts/infer/utils_infer.py CHANGED Viewed

@@ -182,7 +182,7 @@ def initialize_asr_pipeline(device: str = device, dtype=None):
     global asr_pipe
     asr_pipe = pipeline(
         "automatic-speech-recognition",
-        model="openai/whisper-medium",
         torch_dtype=dtype,
         device=device,
     )

     global asr_pipe
     asr_pipe = pipeline(
         "automatic-speech-recognition",
+        model="vinai/PhoWhisper-medium",
         torch_dtype=dtype,
         device=device,
     )