gitea-mirror/F5-TTS
1
0
Fork 0
mirror of https://github.com/SWivid/F5-TTS.git synced 2025-12-12 07:40:43 -08:00
Code Issues Packages Projects Releases Wiki Activity

fix vocoder loading

Browse Source
This commit is contained in:
SWivid
2024-10-30 03:16:09 +08:00
parent da1b40968a
commit 381ea0c82c
4 changed files with 12 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/f5_tts/api.py
View File

@@ -47,7 +47,7 @@ class F5TTS:
self.load_ema_model(model_type, ckpt_file, vocab_file, ode_method, use_ema)
def load_vocoder_model(self, local_path):
self.vocos = load_vocoder(local_path is not None, local_path, self.device)
self.vocoder = load_vocoder(local_path is not None, local_path, self.device)
def load_ema_model(self, model_type, ckpt_file, vocab_file, ode_method, use_ema):
if model_type == "F5-TTS":
@@ -102,6 +102,7 @@ class F5TTS:
ref_text,
gen_text,
self.ema_model,
self.vocoder,
show_info=show_info,
progress=progress,
target_rms=target_rms,