Update download_models.py

nyxieluna · Feb 6, 2024 · 12c44ce · 12c44ce
1 parent 8bfdedc
commit 12c44ce
Showing 1 changed file with 0 additions and 13 deletions.
diff --git a/tools/download_models.py b/tools/download_models.py
@@ -21,11 +21,6 @@ def dl_model(link, model_name, dir_name):
     dl_model(RVC_DOWNLOAD_LINK, "hubert_base.pt", BASE_DIR / "assets/hubert")
     print("Downloading rmvpe.pt...")
     dl_model(RVC_DOWNLOAD_LINK, "rmvpe.pt", BASE_DIR / "assets/rmvpe")
-    print("Downloading vocals.onnx...")
-    dl_model(
-        RVC_DOWNLOAD_LINK + "uvr5_weights/onnx_dereverb_By_FoxJoy/",
-        "vocals.onnx",
-        BASE_DIR / "assets/uvr5_weights/onnx_dereverb_By_FoxJoy",
     )
 
     rvc_models_dir = BASE_DIR / "assets/pretrained"
@@ -58,10 +53,6 @@ def dl_model(link, model_name, dir_name):
         print(f"Downloading {model}...")
         dl_model(RVC_DOWNLOAD_LINK + "pretrained_v2/", model, rvc_models_dir)
 
-    print("Downloading uvr5_weights:")
-
-    rvc_models_dir = BASE_DIR / "assets/uvr5_weights"
-
     model_names = [
         "HP2-%E4%BA%BA%E5%A3%B0vocals%2B%E9%9D%9E%E4%BA%BA%E5%A3%B0instrumentals.pth",
         "HP2_all_vocals.pth",
@@ -71,9 +62,5 @@ def dl_model(link, model_name, dir_name):
         "VR-DeEchoAggressive.pth",
         "VR-DeEchoDeReverb.pth",
         "VR-DeEchoNormal.pth",
-    ]
-    for model in model_names:
-        print(f"Downloading {model}...")
-        dl_model(RVC_DOWNLOAD_LINK + "uvr5_weights/", model, rvc_models_dir)
 
     print("All models downloaded!")