Medix-AI/coqui-tts
Medix-AI
/
coqui-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

This commit is contained in:
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"))}, filecache={"cache_storage": str(get_user_data_dir("tts_cache"))},
mode="rb", mode="rb",
) as f: ) as f:
return torch.load(f, map_location=map_location, **kwargs) return torch.load(f, map_location=map_location, **kwargs, weights_only=False)
else: else:
with fsspec.open(path, "rb") as f: 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( def load_checkpoint(