Update download_models.py
Browse files- download_models.py +4 -56
download_models.py
CHANGED
|
@@ -18,62 +18,10 @@ def dl_model(link, model_name, dir_name):
|
|
| 18 |
|
| 19 |
if __name__ == "__main__":
|
| 20 |
print("Downloading hubert_base.pt...")
|
| 21 |
-
dl_model(RVC_DOWNLOAD_LINK, "hubert_base.pt", BASE_DIR / "assets/hubert")
|
| 22 |
print("Downloading rmvpe.pt...")
|
| 23 |
-
dl_model(RVC_DOWNLOAD_LINK, "rmvpe.pt", BASE_DIR / "assets/rmvpe")
|
| 24 |
-
print("Downloading vocals.onnx...")
|
| 25 |
-
dl_model(
|
| 26 |
-
RVC_DOWNLOAD_LINK + "uvr5_weights/onnx_dereverb_By_FoxJoy/",
|
| 27 |
-
"vocals.onnx",
|
| 28 |
-
BASE_DIR / "assets/uvr5_weights/onnx_dereverb_By_FoxJoy",
|
| 29 |
-
)
|
| 30 |
-
|
| 31 |
-
rvc_models_dir = BASE_DIR / "assets/pretrained"
|
| 32 |
-
|
| 33 |
-
print("Downloading pretrained models:")
|
| 34 |
-
|
| 35 |
-
model_names = [
|
| 36 |
-
"D32k.pth",
|
| 37 |
-
"D40k.pth",
|
| 38 |
-
"D48k.pth",
|
| 39 |
-
"G32k.pth",
|
| 40 |
-
"G40k.pth",
|
| 41 |
-
"G48k.pth",
|
| 42 |
-
"f0D32k.pth",
|
| 43 |
-
"f0D40k.pth",
|
| 44 |
-
"f0D48k.pth",
|
| 45 |
-
"f0G32k.pth",
|
| 46 |
-
"f0G40k.pth",
|
| 47 |
-
"f0G48k.pth",
|
| 48 |
-
]
|
| 49 |
-
for model in model_names:
|
| 50 |
-
print(f"Downloading {model}...")
|
| 51 |
-
dl_model(RVC_DOWNLOAD_LINK + "pretrained/", model, rvc_models_dir)
|
| 52 |
-
|
| 53 |
-
rvc_models_dir = BASE_DIR / "assets/pretrained_v2"
|
| 54 |
-
|
| 55 |
-
print("Downloading pretrained models v2:")
|
| 56 |
-
|
| 57 |
-
for model in model_names:
|
| 58 |
-
print(f"Downloading {model}...")
|
| 59 |
-
dl_model(RVC_DOWNLOAD_LINK + "pretrained_v2/", model, rvc_models_dir)
|
| 60 |
-
|
| 61 |
-
print("Downloading uvr5_weights:")
|
| 62 |
-
|
| 63 |
-
rvc_models_dir = BASE_DIR / "assets/uvr5_weights"
|
| 64 |
-
|
| 65 |
-
model_names = [
|
| 66 |
-
"HP2-%E4%BA%BA%E5%A3%B0vocals%2B%E9%9D%9E%E4%BA%BA%E5%A3%B0instrumentals.pth",
|
| 67 |
-
"HP2_all_vocals.pth",
|
| 68 |
-
"HP3_all_vocals.pth",
|
| 69 |
-
"HP5-%E4%B8%BB%E6%97%8B%E5%BE%8B%E4%BA%BA%E5%A3%B0vocals%2B%E5%85%B6%E4%BB%96instrumentals.pth",
|
| 70 |
-
"HP5_only_main_vocal.pth",
|
| 71 |
-
"VR-DeEchoAggressive.pth",
|
| 72 |
-
"VR-DeEchoDeReverb.pth",
|
| 73 |
-
"VR-DeEchoNormal.pth",
|
| 74 |
-
]
|
| 75 |
-
for model in model_names:
|
| 76 |
-
print(f"Downloading {model}...")
|
| 77 |
-
dl_model(RVC_DOWNLOAD_LINK + "uvr5_weights/", model, rvc_models_dir)
|
| 78 |
|
|
|
|
|
|
|
| 79 |
print("All models downloaded!")
|
|
|
|
| 18 |
|
| 19 |
if __name__ == "__main__":
|
| 20 |
print("Downloading hubert_base.pt...")
|
| 21 |
+
dl_model(RVC_DOWNLOAD_LINK, "hubert_base.pt", BASE_DIR / "assets/rvc/hubert")
|
| 22 |
print("Downloading rmvpe.pt...")
|
| 23 |
+
dl_model(RVC_DOWNLOAD_LINK, "rmvpe.pt", BASE_DIR / "assets/rvc/rmvpe")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
|
| 25 |
+
|
| 26 |
+
|
| 27 |
print("All models downloaded!")
|