57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
import os
|
|
from pathlib import Path
|
|
from huggingface_hub import hf_hub_download
|
|
|
|
now_dir = os.path.dirname(os.path.abspath(__file__))
|
|
BASE_DIR = os.path.join(now_dir, "rvc")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
os.makedirs(os.path.join(BASE_DIR ,"assets","weights"), exist_ok=True)
|
|
weights_path = os.path.join(BASE_DIR ,"assets")
|
|
print("Downloading hubert_base.pt...")
|
|
hf_hub_download(repo_id="lj1995/VoiceConversionWebUI",
|
|
filename="hubert_base.pt",
|
|
subfolder= "",
|
|
local_dir= os.path.join(weights_path, "hubert"))
|
|
print("Downloading rmvpe.pt...")
|
|
hf_hub_download(repo_id="lj1995/VoiceConversionWebUI",
|
|
filename="rmvpe.pt",
|
|
subfolder= "",
|
|
local_dir= os.path.join(weights_path, "rmvpe"))
|
|
|
|
|
|
print("Downloading pretrained models:")
|
|
|
|
model_names = [
|
|
"D40k.pth",
|
|
"D48k.pth",
|
|
"G32k.pth",
|
|
"G40k.pth",
|
|
"G48k.pth",
|
|
"f0D32k.pth",
|
|
"f0D40k.pth",
|
|
"f0D48k.pth",
|
|
"f0G32k.pth",
|
|
"f0G40k.pth",
|
|
"f0G48k.pth",
|
|
]
|
|
for model in model_names:
|
|
print(f"Downloading {model}...")
|
|
hf_hub_download(repo_id="lj1995/VoiceConversionWebUI",
|
|
filename=model,
|
|
subfolder= "pretrained",
|
|
local_dir= weights_path)
|
|
|
|
|
|
print("Downloading pretrained models v2:")
|
|
|
|
for model in model_names:
|
|
print(f"Downloading {model}...")
|
|
hf_hub_download(repo_id="lj1995/VoiceConversionWebUI",
|
|
filename=model,
|
|
subfolder= "pretrained_v2",
|
|
local_dir= weights_path)
|
|
|
|
print("All models downloaded!")
|