|
import os |
|
|
|
from huggingface_hub import hf_hub_download |
|
|
|
|
|
|
|
def check_and_download_files(repo_id, file_list, local_dir): |
|
os.makedirs(local_dir, exist_ok=True) |
|
for file in file_list: |
|
file_path = os.path.join(local_dir, file) |
|
if not os.path.exists(file_path): |
|
print(f"{file} 不存在，从 Hugging Face 仓库下载...") |
|
hf_hub_download( |
|
repo_id=repo_id, |
|
filename=file, |
|
resume_download=True, |
|
local_dir=local_dir, |
|
local_dir_use_symlinks=False, |
|
) |
|
else: |
|
print(f"{file} 已存在，跳过下载。") |
|
|
|
|
|
|
|
repo_id_1 = "fishaudio/fish-speech-1.4" |
|
local_dir_1 = "./checkpoints/fish-speech-1.4" |
|
files_1 = [ |
|
"model.pth", |
|
"README.md", |
|
"special_tokens_map.json", |
|
"tokenizer_config.json", |
|
"tokenizer.json", |
|
"config.json", |
|
"firefly-gan-vq-fsq-8x1024-21hz-generator.pth", |
|
] |
|
|
|
|
|
repo_id_3 = "fishaudio/fish-speech-1" |
|
local_dir_3 = "./" |
|
files_3 = [ |
|
"ffmpeg.exe", |
|
"ffprobe.exe", |
|
] |
|
|
|
|
|
repo_id_4 = "SpicyqSama007/fish-speech-packed" |
|
local_dir_4 = "./" |
|
files_4 = [ |
|
"asr-label-win-x64.exe", |
|
] |
|
|
|
check_and_download_files(repo_id_1, files_1, local_dir_1) |
|
|
|
check_and_download_files(repo_id_3, files_3, local_dir_3) |
|
check_and_download_files(repo_id_4, files_4, local_dir_4) |
|
|
