coqui-ai
/
TTS
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix: Add weights_only=False for PyTorch 2.6+ compatibility

pull/4175/head
gimdonghui 2025-03-21 15:06:22 +09:00
parent dbf1a08a0d
commit 662c0ea783
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
TTS/utils/io.py
View File

@ -48,10 +48,10 @@ def load_fsspec(
filecache={"cache_storage": str(get_user_data_dir("tts_cache"))},
mode="rb",
) as f:
return torch.load(f, map_location=map_location, **kwargs)
return torch.load(f, map_location=map_location, **kwargs, weights_only=False)
else:
with fsspec.open(path, "rb") as f:
return torch.load(f, map_location=map_location, **kwargs)
return torch.load(f, map_location=map_location, **kwargs, weights_only=False)
def load_checkpoint(