update models.py

modules/models.py

@@ -256,8 +256,25 @@ def load_embedder(emb_file: str, emb_name: str):
 
 
 def get_vc_model(model_name: str):
-    model_path = os.path.join(MODELS_DIR, "checkpoints", model_name)
-    weight = torch.load(model_path, map_location="cpu")
+    # model_path = os.path.join(MODELS_DIR, "checkpoints", model_name)
+    # weight = torch.load(model_path, map_location="cpu")
+    
+    # Handle relative paths (e.g., "weights/zet_test1.pth")
+    if "/" in model_name:
+        # It's a relative path, use it directly
+        model_path = os.path.join(ROOT_DIR, model_name)
+        if not os.path.exists(model_path):
+            raise FileNotFoundError(f"Model file not found: {model_path}")
+    else:
+        # It's just a filename, check in weights folder first (for custom models)
+        weights_path = os.path.join(ROOT_DIR, "weights", model_name)
+        if os.path.exists(weights_path):
+            model_path = weights_path
+        else:
+            # Fallback to checkpoints folder
+            model_path = os.path.join(MODELS_DIR, "checkpoints", model_name)
+    
+    weight = torch.load(model_path, map_location="cpu", weights_only=False)
     return VoiceConvertModel(model_name, weight)