97 lines
3.0 KiB
Python
97 lines
3.0 KiB
Python
import os
|
|
import wget
|
|
|
|
url_base = "https://huggingface.co/nolinkhere"
|
|
pretraineds_v1_list = [
|
|
(
|
|
"pretrained_v1/",
|
|
[
|
|
"D32k.pth",
|
|
"D40k.pth",
|
|
"D48k.pth",
|
|
"G32k.pth",
|
|
"G40k.pth",
|
|
"G48k.pth",
|
|
"f0D32k.pth",
|
|
"f0D40k.pth",
|
|
"f0D48k.pth",
|
|
"f0G32k.pth",
|
|
"f0G40k.pth",
|
|
"f0G48k.pth",
|
|
],
|
|
),
|
|
]
|
|
pretraineds_v2_list = [
|
|
(
|
|
"pretrained_v2/",
|
|
[
|
|
"D32k.pth",
|
|
"D40k.pth",
|
|
"D48k.pth",
|
|
"G32k.pth",
|
|
"G40k.pth",
|
|
"G48k.pth",
|
|
"f0D32k.pth",
|
|
"f0D40k.pth",
|
|
"f0D48k.pth",
|
|
"f0G32k.pth",
|
|
"f0G40k.pth",
|
|
"f0G48k.pth",
|
|
],
|
|
),
|
|
]
|
|
|
|
models_list = [
|
|
"hubert_base.pt",
|
|
"contentvec_base.pt",
|
|
"rmvpe.pt",
|
|
"fcpe.pt",
|
|
# "rmvpe.onnx"
|
|
]
|
|
|
|
executables_list = ["ffmpeg.exe", "ffprobe.exe"]
|
|
|
|
folder_mapping_list = {
|
|
"pretrained_v1/": "rvc/pretraineds/pretrained_v1/",
|
|
"pretrained_v2/": "rvc/pretraineds/pretrained_v2/",
|
|
}
|
|
|
|
|
|
def prequisites_download_pipeline(pretraineds_v1, pretraineds_v2, models, exe):
|
|
def download_files(file_list):
|
|
for file_name in file_list:
|
|
destination_path = os.path.join(file_name)
|
|
url = f"{url_base}/{file_name}"
|
|
if not os.path.exists(destination_path):
|
|
os.makedirs(os.path.dirname(destination_path) or ".", exist_ok=True)
|
|
print(f"\nDownloading {url} to {destination_path}...")
|
|
wget.download(url, out=destination_path)
|
|
|
|
if models == "True":
|
|
download_files(models_list)
|
|
|
|
if exe == "True" and os.name == "nt":
|
|
download_files(executables_list)
|
|
|
|
if pretraineds_v1 == "True":
|
|
for remote_folder, file_list in pretraineds_v1_list:
|
|
local_folder = folder_mapping_list.get(remote_folder, "")
|
|
for file in file_list:
|
|
destination_path = os.path.join(local_folder, file)
|
|
url = f"{url_base}/{remote_folder}{file}"
|
|
if not os.path.exists(destination_path):
|
|
os.makedirs(os.path.dirname(destination_path) or ".", exist_ok=True)
|
|
print(f"\nDownloading {url} to {destination_path}...")
|
|
wget.download(url, out=destination_path)
|
|
|
|
if pretraineds_v2 == "True":
|
|
for remote_folder, file_list in pretraineds_v2_list:
|
|
local_folder = folder_mapping_list.get(remote_folder, "")
|
|
for file in file_list:
|
|
destination_path = os.path.join(local_folder, file)
|
|
url = f"{url_base}/{remote_folder}{file}"
|
|
if not os.path.exists(destination_path):
|
|
os.makedirs(os.path.dirname(destination_path) or ".", exist_ok=True)
|
|
print(f"\nDownloading {url} to {destination_path}...")
|
|
wget.download(url, out=destination_path)
|