From 662c0ea7831dbd0af6eceeb787504ed9e92eeaeb Mon Sep 17 00:00:00 2001 From: gimdonghui <00_kdh@naver.com> Date: Fri, 21 Mar 2025 15:06:22 +0900 Subject: [PATCH] Fix: Add weights_only=False for PyTorch 2.6+ compatibility --- TTS/utils/io.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/TTS/utils/io.py b/TTS/utils/io.py index 3107ba66..f80f4ee2 100644 --- a/TTS/utils/io.py +++ b/TTS/utils/io.py @@ -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(