若干杂项,界面优化 (#1388)
现支持直接启动inference_webui,支持选择version和i18n 将不同版本模型放在两个文件夹中,启动inference_webui后将根据version选择模型文件夹 训练保存文件夹也将根据verison变化
This commit is contained in:
17
webui.py
17
webui.py
@@ -1,6 +1,7 @@
|
||||
import os,shutil,sys,pdb,re
|
||||
version="v2"if sys.argv[0]=="v2" else"v1"
|
||||
language=sys.argv[-1] if sys.argv[-1]!='v2' and sys.argv[-1]!='v1' else 'auto'
|
||||
if len(sys.argv)==1:sys.argv.append('v1')
|
||||
version="v2"if sys.argv[1]=="v2" else"v1"
|
||||
language=sys.argv[-1] if len(sys.argv[-1])==5 else "auto"
|
||||
os.environ["version"]=version
|
||||
os.environ["language"]=language
|
||||
now_dir = os.getcwd()
|
||||
@@ -121,10 +122,12 @@ def get_weights_names():
|
||||
for name in os.listdir(GPT_weight_root):
|
||||
if name.endswith(".ckpt"): GPT_names.append(name)
|
||||
return SoVITS_names,GPT_names
|
||||
SoVITS_weight_root="SoVITS_weights"
|
||||
GPT_weight_root="GPT_weights"
|
||||
os.makedirs(SoVITS_weight_root,exist_ok=True)
|
||||
os.makedirs(GPT_weight_root,exist_ok=True)
|
||||
SoVITS_weight_root="SoVITS_weights_v2" if version=='v2' else "SoVITS_weights"
|
||||
GPT_weight_root="GPT_weights_v2" if version=='v2' else "GPT_weights"
|
||||
os.makedirs("SoVITS_weights",exist_ok=True)
|
||||
os.makedirs("GPT_weights",exist_ok=True)
|
||||
os.makedirs("SoVITS_weights_v2",exist_ok=True)
|
||||
os.makedirs("GPT_weights_v2",exist_ok=True)
|
||||
SoVITS_names,GPT_names = get_weights_names()
|
||||
|
||||
def custom_sort_key(s):
|
||||
@@ -208,7 +211,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)
|
||||
|
||||
Reference in New Issue
Block a user