mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-10 10:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix infer error

Browse Source
This commit is contained in:
root 2024-12-25 02:17:18 +00:00
parent 03d500a414
commit f3fca0c81b
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
egs/wenetspeech4tts/TTS/f5-tts/infer.py
View File

@ -281,9 +281,8 @@ def main():
vocoder = vocoder.eval().to(device) vocoder = vocoder.eval().to(device)
model = get_model(args).eval().to(device) model = get_model(args).eval().to(device)
checkpoint = torch.load(args.model_path, map_location="cpu", weights_only=True) checkpoint = torch.load(args.model_path, map_location="cpu")
if "ema_model_state_dict" in checkpoint or 'model_state_dict' in checkpoint:
if "model_state_dict" or "ema_model_state_dict" in checkpoint:
model = load_F5_TTS_pretrained_checkpoint(model, args.model_path) model = load_F5_TTS_pretrained_checkpoint(model, args.model_path)
else: else:
_ = load_checkpoint( _ = load_checkpoint(