Merge pull request #8 from shavit/torch-load-weights-only

Load weights only in torch.load

.github/workflows/tests.yml

@@ -39,6 +39,7 @@ jobs:
       - name: Upload coverage data
         uses: actions/upload-artifact@v4
         with:
+          include-hidden-files: true
           name: coverage-data-${{ matrix.python-version }}-${{ matrix.uv-resolution }}
           path: .coverage.*
           if-no-files-found: ignore

pyproject.toml

@@ -7,7 +7,7 @@ include = ["trainer*"]
 
 [project]
 name = "coqui-tts-trainer"
-version = "0.1.4"
+version = "0.1.5"
 description = "General purpose model trainer for PyTorch that is more flexible than it should be, by 🐸Coqui."
 readme = "README.md"
 requires-python = ">=3.9, <3.13"

trainer/io.py

@@ -77,10 +77,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, weights_only=True, **kwargs)
     else:
         with fsspec.open(str(path), "rb") as f:
-            return torch.load(f, map_location=map_location, **kwargs)
+            return torch.load(f, map_location=map_location, weights_only=True, **kwargs)
 
 
 def load_checkpoint(