From 252a90a0f619d819a31cf9c2f9cc5e09b2e41193 Mon Sep 17 00:00:00 2001
From: VSlobolinskyi <vitalii.slobolinskyi@gmail.com>
Date: Tue, 18 Mar 2025 00:55:14 +0200
Subject: [PATCH] Adjust download script

---
 tools/{download_models.py => download_assets.py} | 6 ++++++
 1 file changed, 6 insertions(+)
 rename tools/{download_models.py => download_assets.py} (88%)

diff --git a/tools/download_models.py b/tools/download_assets.py
similarity index 88%
rename from tools/download_models.py
rename to tools/download_assets.py
index 94e0389..63f0220 100644
--- a/tools/download_models.py
+++ b/tools/download_assets.py
@@ -21,12 +21,18 @@ if __name__ == "__main__":
     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 rmvpe.pt...")
+    dl_model(RVC_DOWNLOAD_LINK, "rmvpe.onnx", 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",
     )
+    dl_model(RVC_DOWNLOAD_LINK, "ffprobe.exe", BASE_DIR / "./")
+    print("Downloading ffprobe.exe...")
+    dl_model(RVC_DOWNLOAD_LINK, "ffmpeg.exe", BASE_DIR / "./")
+    print("Downloading ffmpeg.exe...")
 
     rvc_models_dir = BASE_DIR / "assets/pretrained"