gitea-mirror/F5-TTS
1
0
Fork 0
mirror of https://github.com/SWivid/F5-TTS.git synced 2026-01-05 09:39:52 -08:00
Code Issues Packages Projects Releases Wiki Activity

default to weights_only=True for safer loading

Browse Source
This commit is contained in:
Jarod Mica
2024-10-15 00:37:46 -07:00
parent 49b465f5d8
commit 31e5051d51
4 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/eval_infer_batch.py
View File

@@ -127,7 +127,7 @@ local = False
if local:
vocos_local_path = "../checkpoints/charactr/vocos-mel-24khz"
vocos = Vocos.from_hparams(f"{vocos_local_path}/config.yaml")
state_dict = torch.load(f"{vocos_local_path}/pytorch_model.bin", map_location=device)
state_dict = torch.load(f"{vocos_local_path}/pytorch_model.bin", weights_only=True, map_location=device)
vocos.load_state_dict(state_dict)
vocos.eval()
else: