This commit is contained in:
XXXXRT666 2024-08-03 12:27:52 +08:00
parent f1c29f3e5b
commit 9de98a1d49
2 changed files with 4 additions and 2 deletions

View File

@ -20,6 +20,8 @@ import torch
version=os.environ.get("version","v1")
language=os.environ.get("language","auto")
version="v2"if sys.argv[1]=="v2" else version
language=sys.argv[-1] if sys.argv[-1]!='v2' and sys.argv[-1]!='v1' else language
pretrained_sovits_name="GPT_SoVITS/pretrained_models/s2G488k.pth"if version=="v1"else"GPT_SoVITS/pretrained_models/gsv-v2final-pretrained/s2G2333k.pth"
pretrained_gpt_name="GPT_SoVITS/pretrained_models/s1bert25hz-2kh-longer-epoch=68e-step=50232.ckpt"if version=="v1"else "GPT_SoVITS/pretrained_models/gsv-v2final-pretrained/s1bert25hz-5kh-longer-epoch=12-step=369668.ckpt"

View File

@ -1,5 +1,5 @@
import os,shutil,sys,pdb,re
version="v2"if sys.argv[0]=="v2" else"v1"
version="v2"if sys.argv[1]=="v2" else"v1"
language=sys.argv[-1] if sys.argv[-1]!='v2' and sys.argv[-1]!='v1' else 'auto'
os.environ["version"]=version
os.environ["language"]=language
@ -208,7 +208,7 @@ def change_tts_inference(if_tts,bert_path,cnhubert_base_path,gpu_number,gpt_path
os.environ["is_half"]=str(is_half)
os.environ["infer_ttswebui"]=str(webui_port_infer_tts)
os.environ["is_share"]=str(is_share)
cmd = '"%s" GPT_SoVITS/inference_webui.py'%(python_exec)
cmd = '"%s" GPT_SoVITS/inference_webui.py "%s"'%(python_exec, language)
yield i18n("TTS推理进程已开启")
print(cmd)
p_tts_inference = Popen(cmd, shell=True)