Update io.py

io.py

@@ -30,17 +30,10 @@ def load_fsspec(
     cache: bool = True,
     **kwargs,
 ) -> Any:
-    """Like torch.load but can load from other locations (e.g. s3:// , gs://).
+    """Like torch.load but can load from other locations (e.g. s3:// , gs://)."""
+    if "weights_only" not in kwargs:
+        kwargs["weights_only"] = False  # 👈 forzar compatibilidad con checkpoints antiguos
 
-    Args:
-        path: Any path or url supported by fsspec.
-        map_location: torch.device or str.
-        cache: If True, cache a remote file locally for subsequent calls. It is cached under `get_user_data_dir()/tts_cache`. Defaults to True.
-        **kwargs: Keyword arguments forwarded to torch.load.
-
-    Returns:
-        Object stored in path.
-    """
     is_local = os.path.isdir(path) or os.path.isfile(path)
     if cache and not is_local:
         with fsspec.open(
@@ -53,7 +46,6 @@ def load_fsspec(
         with fsspec.open(path, "rb") as f:
             return torch.load(f, map_location=map_location, **kwargs)
 
-
 def load_checkpoint(
     model, checkpoint_path, use_cuda=False, eval=False, cache=False
 ):  # pylint: disable=redefined-builtin