From d2e4062607aef51e9b82be2c9989f3f19e5d0830 Mon Sep 17 00:00:00 2001
From: shavit <shavit@users.noreply.github.com>
Date: Sat, 31 Aug 2024 06:15:24 -0400
Subject: [PATCH 1/3] fix(io): load weights only in torch.load

---
 trainer/io.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/trainer/io.py b/trainer/io.py
index 165f6ea..615ad0a 100644
--- a/trainer/io.py
+++ b/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(

From 3f38e1ce21f3c4e131ff04fe732adbb0530285db Mon Sep 17 00:00:00 2001
From: Enno Hermann <enno.hermann@idiap.ch>
Date: Thu, 12 Sep 2024 16:17:13 +0200
Subject: [PATCH 2/3] ci: explicitly upload hidden files for coverage

Due to breaking change in upload-artifact action:
https://github.com/actions/upload-artifact/issues/602
---
 .github/workflows/tests.yml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 64cd11d..441941b 100644
--- a/.github/workflows/tests.yml
+++ b/.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

From f796733d70bcc7c936777b3248e6962007b30d60 Mon Sep 17 00:00:00 2001
From: Enno Hermann <enno.hermann@idiap.ch>
Date: Thu, 12 Sep 2024 15:50:16 +0200
Subject: [PATCH 3/3] chore: update version to 0.1.5

---
 pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index ebb1afc..46fc7c2 100644
--- a/pyproject.toml
+++ b/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"