This commit is contained in:
Donghee Kim 2025-03-21 15:11:15 +09:00 committed by GitHub
commit d1c096b294
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions

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(