blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-04-05 04:22:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: docker add v2 models

Browse Source
This commit is contained in:
Zone Tome 2024-10-30 09:06:02 +00:00
parent 98cc47699c
commit 09f28ae518
2 changed files with 47 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
Docker/download.py
View File

@ -3,3 +3,44 @@ from modelscope import snapshot_download
model_dir = snapshot_download('damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch',revision="v2.0.4")
model_dir = snapshot_download('damo/speech_fsmn_vad_zh-cn-16k-common-pytorch',revision="v2.0.4")
model_dir = snapshot_download('damo/punc_ct-transformer_zh-cn-common-vocab272727-pytorch',revision="v2.0.4")
import nltk
nltk.download('averaged_perceptron_tagger_eng')
# Download https://paddlespeech.bj.bcebos.com/Parakeet/released_models/g2p/G2PWModel_1.1.zip unzip and rename to G2PWModel, and then place them in GPT_SoVITS/text.
import os
import zipfile
import shutil
import requests
# 获取当前文件的路径
current_file_path = os.path.abspath(__file__)
current_dir = os.path.dirname(current_file_path)
# 定义下载链接和目标路径
url = 'https://paddlespeech.bj.bcebos.com/Parakeet/released_models/g2p/G2PWModel_1.1.zip'
download_path = os.path.join(current_dir, 'G2PWModel_1.1.zip')
target_dir = os.path.join(current_dir, '../GPT_SoVITS/text/')
# 下载文件
response = requests.get(url)
with open(download_path, 'wb') as file:
file.write(response.content)
# 解压文件
with zipfile.ZipFile(download_path, 'r') as zip_ref:
zip_ref.extractall(current_dir)
# 重命名解压后的文件夹
os.rename(os.path.join(current_dir, 'G2PWModel_1.1'), os.path.join(current_dir, 'G2PWModel'))
# 移动文件夹到目标目录
if not os.path.exists(target_dir):
os.makedirs(target_dir)
shutil.move(os.path.join(current_dir, 'G2PWModel'), target_dir)
# 清理临时文件
os.remove(download_path)
print("Download G2PWModel successfully")

6
Docker/links.txt
View File

@ -17,6 +17,12 @@ https://huggingface.co/lj1995/GPT-SoVITS/resolve/main/chinese-roberta-wwm-ext-la
out=GPT_SoVITS/pretrained_models/chinese-roberta-wwm-ext-large/pytorch_model.bin
https://huggingface.co/lj1995/GPT-SoVITS/resolve/main/chinese-roberta-wwm-ext-large/tokenizer.json
out=GPT_SoVITS/pretrained_models/chinese-roberta-wwm-ext-large/tokenizer.json
https://huggingface.co/lj1995/GPT-SoVITS/resolve/main/gsv-v2final-pretrained/s1bert25hz-5kh-longer-epoch=12-step=369668.ckpt
out=GPT_SoVITS/pretrained_models/gsv-v2final-pretrained/s1bert25hz-5kh-longer-epoch=12-step=369668.ckpt
https://huggingface.co/lj1995/GPT-SoVITS/resolve/main/gsv-v2final-pretrained/s2D2333k.pth
out=GPT_SoVITS/pretrained_models/gsv-v2final-pretrained/s2D2333k.pth
https://huggingface.co/lj1995/GPT-SoVITS/resolve/main/gsv-v2final-pretrained/s2G2333k.pth
out=GPT_SoVITS/pretrained_models/gsv-v2final-pretrained/s2G2333k.pth
# UVR5
https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main/uvr5_weights/HP2_all_vocals.pth
out=tools/uvr5/uvr5_weights/HP2_all_vocals.pth