hynt commited on
Commit
0dc060d
·
verified ·
1 Parent(s): fff0f9b

Update f5_tts/infer/utils_infer.py

Browse files
Files changed (1) hide show
  1. f5_tts/infer/utils_infer.py +1 -1
f5_tts/infer/utils_infer.py CHANGED
@@ -182,7 +182,7 @@ def initialize_asr_pipeline(device: str = device, dtype=None):
182
  global asr_pipe
183
  asr_pipe = pipeline(
184
  "automatic-speech-recognition",
185
- model="openai/whisper-medium",
186
  torch_dtype=dtype,
187
  device=device,
188
  )
 
182
  global asr_pipe
183
  asr_pipe = pipeline(
184
  "automatic-speech-recognition",
185
+ model="vinai/PhoWhisper-medium",
186
  torch_dtype=dtype,
187
  device=device,
188
  )