more torch.load fixes

IAHispano · Feb 1, 2025 · eb21d9d · eb21d9d
1 parent c5c4bf4
commit eb21d9d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/rvc/train/train.py b/rvc/train/train.py
@@ -134,7 +134,7 @@ def record(self):
 
 
 def verify_checkpoint_shapes(checkpoint_path, model):
-    checkpoint = torch.load(checkpoint_path, map_location="cpu")
+    checkpoint = torch.load(checkpoint_path, map_location="cpu", weights_only=True)
     checkpoint_state_dict = checkpoint["model"]
     try:
         if hasattr(model, "module"):

diff --git a/tabs/inference/inference.py b/tabs/inference/inference.py
@@ -334,7 +334,7 @@ def refresh_embedders_folders():
 def get_speakers_id(model):
     if model:
         try:
-            model_data = torch.load(os.path.join(now_dir, model), map_location="cpu")
+            model_data = torch.load(os.path.join(now_dir, model), map_location="cpu", weights_only=True)
             speakers_id = model_data.get("speakers_id")
             if speakers_id:
                 return list(range(speakers_id))