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

Reformat scripts

Browse Source
This commit is contained in:
laubonghaudoi 2025-03-19 11:39:03 -07:00
parent c18c9a9eed
commit d32d70c2b1
3 changed files with 517 additions and 415 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@ -18,4 +18,3 @@ TEMP
weight.json weight.json
ffmpeg* ffmpeg*
ffprobe* ffprobe*
zoengjyutgaai*

2
GPT_SoVITS/prepare_datasets/1-get-text.py
View File

@ -8,8 +8,8 @@ from time import time as ttime
import torch import torch
from text.cleaner import clean_text from text.cleaner import clean_text
from transformers import AutoModelForMaskedLM, AutoTokenizer
from tqdm import tqdm from tqdm import tqdm
from transformers import AutoModelForMaskedLM, AutoTokenizer
from tools.my_utils import clean_path from tools.my_utils import clean_path

209
webui.py
View File

@ -1,15 +1,45 @@
import os,sys import json
if len(sys.argv)==1:sys.argv.append('v2') import os
import platform
import re
import shutil
import signal
import site
import subprocess
import sys
import traceback
import warnings
from multiprocessing import cpu_count
from subprocess import Popen
import gradio as gr
import psutil
import torch
import yaml
from config import (
exp_root,
infer_device,
is_half,
is_share,
python_exec,
webui_port_infer_tts,
webui_port_main,
webui_port_subfix,
webui_port_uvr5,
)
from tools import my_utils
from tools.asr.config import asr_dict
from tools.i18n.i18n import I18nAuto, scan_language_list
from tools.my_utils import check_details, check_for_existance
if len(sys.argv) == 1:
sys.argv.append('v2')
version = "v1"if sys.argv[1] == "v1" else "v2" version = "v1"if sys.argv[1] == "v1" else "v2"
os.environ["version"] = version os.environ["version"] = version
now_dir = os.getcwd() now_dir = os.getcwd()
sys.path.insert(0, now_dir) sys.path.insert(0, now_dir)
import warnings
warnings.filterwarnings("ignore") warnings.filterwarnings("ignore")
import json,yaml,torch,pdb,re,shutil
import platform
import psutil
import signal
os.environ['TORCH_DISTRIBUTED_DEBUG'] = 'INFO' os.environ['TORCH_DISTRIBUTED_DEBUG'] = 'INFO'
torch.manual_seed(233333) torch.manual_seed(233333)
tmp = os.path.join(now_dir, "TEMP") tmp = os.path.join(now_dir, "TEMP")
@ -17,7 +47,8 @@ os.makedirs(tmp, exist_ok=True)
os.environ["TEMP"] = tmp os.environ["TEMP"] = tmp
if (os.path.exists(tmp)): if (os.path.exists(tmp)):
for name in os.listdir(tmp): for name in os.listdir(tmp):
if(name=="jieba.cache"):continue if (name == "jieba.cache"):
continue
path = "%s/%s" % (tmp, name) path = "%s/%s" % (tmp, name)
delete = os.remove if os.path.isfile(path) else shutil.rmtree delete = os.remove if os.path.isfile(path) else shutil.rmtree
try: try:
@ -25,13 +56,12 @@ if(os.path.exists(tmp)):
except Exception as e: except Exception as e:
print(str(e)) print(str(e))
pass pass
import site
import traceback
site_packages_roots = [] site_packages_roots = []
for path in site.getsitepackages(): for path in site.getsitepackages():
if "packages" in path: if "packages" in path:
site_packages_roots.append(path) site_packages_roots.append(path)
if(site_packages_roots==[]):site_packages_roots=["%s/runtime/Lib/site-packages" % now_dir] if (site_packages_roots == []):
site_packages_roots = ["%s/runtime/Lib/site-packages" % now_dir]
# os.environ["OPENBLAS_NUM_THREADS"] = "4" # os.environ["OPENBLAS_NUM_THREADS"] = "4"
os.environ["no_proxy"] = "localhost, 127.0.0.1, ::1" os.environ["no_proxy"] = "localhost, 127.0.0.1, ::1"
os.environ["all_proxy"] = "" os.environ["all_proxy"] = ""
@ -45,28 +75,17 @@ for site_packages_root in site_packages_roots:
% (now_dir, now_dir, now_dir, now_dir, now_dir, now_dir) % (now_dir, now_dir, now_dir, now_dir, now_dir, now_dir)
) )
break break
except PermissionError as e: except PermissionError:
traceback.print_exc() traceback.print_exc()
from tools import my_utils
import shutil
import pdb
import subprocess
from subprocess import Popen
import signal
from config import python_exec,infer_device,is_half,exp_root,webui_port_main,webui_port_infer_tts,webui_port_uvr5,webui_port_subfix,is_share
from tools.i18n.i18n import I18nAuto, scan_language_list
language = sys.argv[-1] if sys.argv[-1] in scan_language_list() else "Auto" language = sys.argv[-1] if sys.argv[-1] in scan_language_list() else "Auto"
os.environ["language"] = language os.environ["language"] = language
i18n = I18nAuto(language=language) i18n = I18nAuto(language=language)
from scipy.io import wavfile
from tools.my_utils import load_audio, check_for_existance, check_details
from multiprocessing import cpu_count
# os.environ['PYTORCH_ENABLE_MPS_FALLBACK'] = '1' # 当遇到mps不支持的步骤时使用cpu # os.environ['PYTORCH_ENABLE_MPS_FALLBACK'] = '1' # 当遇到mps不支持的步骤时使用cpu
try: try:
import gradio.analytics as analytics import gradio.analytics as analytics
analytics.version_check = lambda: None analytics.version_check = lambda: None
except:... except:
import gradio as gr ...
n_cpu = cpu_count() n_cpu = cpu_count()
ngpu = torch.cuda.device_count() ngpu = torch.cuda.device_count()
@ -92,6 +111,7 @@ if torch.cuda.is_available() or ngpu != 0:
# gpu_infos.append("%s\t%s" % ("0", "Apple GPU")) # gpu_infos.append("%s\t%s" % ("0", "Apple GPU"))
# mem.append(psutil.virtual_memory().total/ 1024 / 1024 / 1024) # 实测使用系统内存作为显存不会爆显存 # mem.append(psutil.virtual_memory().total/ 1024 / 1024 / 1024) # 实测使用系统内存作为显存不会爆显存
def set_default(): def set_default():
global default_batch_size, default_max_batch_size, gpu_info, default_sovits_epoch, default_sovits_save_every_epoch, max_sovits_epoch, max_sovits_save_every_epoch, default_batch_size_s1, if_force_ckpt global default_batch_size, default_max_batch_size, gpu_info, default_sovits_epoch, default_sovits_save_every_epoch, max_sovits_epoch, max_sovits_save_every_epoch, default_batch_size_s1, if_force_ckpt
if_force_ckpt = False if_force_ckpt = False
@ -139,23 +159,32 @@ def set_default():
default_batch_size_s1 = max(1, default_batch_size_s1) default_batch_size_s1 = max(1, default_batch_size_s1)
default_max_batch_size = default_batch_size * 3 default_max_batch_size = default_batch_size * 3
set_default() set_default()
gpus = "-".join([i[0] for i in gpu_infos]) gpus = "-".join([i[0] for i in gpu_infos])
default_gpu_numbers = str(sorted(list(set_gpu_numbers))[0]) default_gpu_numbers = str(sorted(list(set_gpu_numbers))[0])
def fix_gpu_number(input): # 将越界的number强制改到界内 def fix_gpu_number(input): # 将越界的number强制改到界内
try: try:
if(int(input)not in set_gpu_numbers):return default_gpu_numbers if (int(input)not in set_gpu_numbers):
except:return input return default_gpu_numbers
except:
return input return input
return input
def fix_gpu_numbers(inputs): def fix_gpu_numbers(inputs):
output = [] output = []
try: try:
for input in inputs.split(","):output.append(str(fix_gpu_number(input))) for input in inputs.split(","):
output.append(str(fix_gpu_number(input)))
return ",".join(output) return ",".join(output)
except: except:
return inputs return inputs
pretrained_sovits_name = ["GPT_SoVITS/pretrained_models/s2G488k.pth", "GPT_SoVITS/pretrained_models/gsv-v2final-pretrained/s2G2333k.pth", "GPT_SoVITS/pretrained_models/s2Gv3.pth"] pretrained_sovits_name = ["GPT_SoVITS/pretrained_models/s2G488k.pth", "GPT_SoVITS/pretrained_models/gsv-v2final-pretrained/s2G2333k.pth", "GPT_SoVITS/pretrained_models/s2Gv3.pth"]
pretrained_gpt_name = ["GPT_SoVITS/pretrained_models/s1bert25hz-2kh-longer-epoch=68e-step=50232.ckpt", "GPT_SoVITS/pretrained_models/gsv-v2final-pretrained/s1bert25hz-5kh-longer-epoch=12-step=369668.ckpt", "GPT_SoVITS/pretrained_models/s1v3.ckpt"] pretrained_gpt_name = ["GPT_SoVITS/pretrained_models/s1bert25hz-2kh-longer-epoch=68e-step=50232.ckpt", "GPT_SoVITS/pretrained_models/gsv-v2final-pretrained/s1bert25hz-5kh-longer-epoch=12-step=369668.ckpt", "GPT_SoVITS/pretrained_models/s1v3.ckpt"]
@ -170,31 +199,41 @@ if _:
_ = [[], []] _ = [[], []]
for i in range(3): for i in range(3):
if os.path.exists(pretrained_gpt_name[i]):_[0].append(pretrained_gpt_name[i]) if os.path.exists(pretrained_gpt_name[i]):
else:_[0].append("")##没有下pretrained模型的，说不定他们是想自己从零训底模呢 _[0].append(pretrained_gpt_name[i])
if os.path.exists(pretrained_sovits_name[i]):_[-1].append(pretrained_sovits_name[i]) else:
else:_[-1].append("") _[0].append("") # 没有下pretrained模型的，说不定他们是想自己从零训底模呢
if os.path.exists(pretrained_sovits_name[i]):
_[-1].append(pretrained_sovits_name[i])
else:
_[-1].append("")
pretrained_gpt_name, pretrained_sovits_name = _ pretrained_gpt_name, pretrained_sovits_name = _
SoVITS_weight_root = ["SoVITS_weights", "SoVITS_weights_v2", "SoVITS_weights_v3"] SoVITS_weight_root = ["SoVITS_weights", "SoVITS_weights_v2", "SoVITS_weights_v3"]
GPT_weight_root = ["GPT_weights", "GPT_weights_v2", "GPT_weights_v3"] GPT_weight_root = ["GPT_weights", "GPT_weights_v2", "GPT_weights_v3"]
for root in SoVITS_weight_root + GPT_weight_root: for root in SoVITS_weight_root + GPT_weight_root:
os.makedirs(root, exist_ok=True) os.makedirs(root, exist_ok=True)
def get_weights_names(): def get_weights_names():
SoVITS_names = [name for name in pretrained_sovits_name if name != ""] SoVITS_names = [name for name in pretrained_sovits_name if name != ""]
for path in SoVITS_weight_root: for path in SoVITS_weight_root:
for name in os.listdir(path): for name in os.listdir(path):
if name.endswith(".pth"): SoVITS_names.append("%s/%s" % (path, name)) if name.endswith(".pth"):
SoVITS_names.append("%s/%s" % (path, name))
GPT_names = [name for name in pretrained_gpt_name if name != ""] GPT_names = [name for name in pretrained_gpt_name if name != ""]
for path in GPT_weight_root: for path in GPT_weight_root:
for name in os.listdir(path): for name in os.listdir(path):
if name.endswith(".ckpt"): GPT_names.append("%s/%s" % (path, name)) if name.endswith(".ckpt"):
GPT_names.append("%s/%s" % (path, name))
return SoVITS_names, GPT_names return SoVITS_names, GPT_names
SoVITS_names, GPT_names = get_weights_names() SoVITS_names, GPT_names = get_weights_names()
for path in SoVITS_weight_root + GPT_weight_root: for path in SoVITS_weight_root + GPT_weight_root:
os.makedirs(path, exist_ok=True) os.makedirs(path, exist_ok=True)
def custom_sort_key(s): def custom_sort_key(s):
# 使用正则表达式提取字符串中的数字部分和非数字部分 # 使用正则表达式提取字符串中的数字部分和非数字部分
parts = re.split('(\d+)', s) parts = re.split('(\d+)', s)
@ -202,16 +241,19 @@ def custom_sort_key(s):
parts = [int(part) if part.isdigit() else part for part in parts] parts = [int(part) if part.isdigit() else part for part in parts]
return parts return parts
def change_choices(): def change_choices():
SoVITS_names, GPT_names = get_weights_names() SoVITS_names, GPT_names = get_weights_names()
return {"choices": sorted(SoVITS_names, key=custom_sort_key), "__type__": "update"}, {"choices": sorted(GPT_names, key=custom_sort_key), "__type__": "update"} return {"choices": sorted(SoVITS_names, key=custom_sort_key), "__type__": "update"}, {"choices": sorted(GPT_names, key=custom_sort_key), "__type__": "update"}
p_label = None p_label = None
p_uvr5 = None p_uvr5 = None
p_asr = None p_asr = None
p_denoise = None p_denoise = None
p_tts_inference = None p_tts_inference = None
def kill_proc_tree(pid, including_parent=True): def kill_proc_tree(pid, including_parent=True):
try: try:
parent = psutil.Process(pid) parent = psutil.Process(pid)
@ -231,7 +273,10 @@ def kill_proc_tree(pid, including_parent=True):
except OSError: except OSError:
pass pass
system = platform.system() system = platform.system()
def kill_process(pid, process_name=""): def kill_process(pid, process_name=""):
if (system == "Windows"): if (system == "Windows"):
cmd = "taskkill /t /f /pid %s" % pid cmd = "taskkill /t /f /pid %s" % pid
@ -241,6 +286,7 @@ def kill_process(pid, process_name=""):
kill_proc_tree(pid) kill_proc_tree(pid)
print(process_name + i18n("进程已终止")) print(process_name + i18n("进程已终止"))
def process_info(process_name="", indicator=""): def process_info(process_name="", indicator=""):
if indicator == "opened": if indicator == "opened":
return process_name + i18n("已开启") return process_name + i18n("已开启")
@ -263,7 +309,10 @@ def process_info(process_name="", indicator=""):
else: else:
return process_name return process_name
process_name_subfix = i18n("音频标注WebUI") process_name_subfix = i18n("音频标注WebUI")
def change_label(path_list): def change_label(path_list):
global p_label global p_label
if p_label is None: if p_label is None:
@ -278,7 +327,10 @@ def change_label(path_list):
p_label = None p_label = None
yield process_info(process_name_subfix, "closed"), {'__type__': 'update', 'visible': True}, {'__type__': 'update', 'visible': False} yield process_info(process_name_subfix, "closed"), {'__type__': 'update', 'visible': True}, {'__type__': 'update', 'visible': False}
process_name_uvr5 = i18n("人声分离WebUI") process_name_uvr5 = i18n("人声分离WebUI")
def change_uvr5(): def change_uvr5():
global p_uvr5 global p_uvr5
if p_uvr5 is None: if p_uvr5 is None:
@ -291,14 +343,17 @@ def change_uvr5():
p_uvr5 = None p_uvr5 = None
yield process_info(process_name_uvr5, "closed"), {'__type__': 'update', 'visible': True}, {'__type__': 'update', 'visible': False} yield process_info(process_name_uvr5, "closed"), {'__type__': 'update', 'visible': True}, {'__type__': 'update', 'visible': False}
process_name_tts = i18n("TTS推理WebUI") process_name_tts = i18n("TTS推理WebUI")
def change_tts_inference(bert_path, cnhubert_base_path, gpu_number, gpt_path, sovits_path, batched_infer_enabled): def change_tts_inference(bert_path, cnhubert_base_path, gpu_number, gpt_path, sovits_path, batched_infer_enabled):
global p_tts_inference global p_tts_inference
if batched_infer_enabled: if batched_infer_enabled:
cmd = '"%s" GPT_SoVITS/inference_webui_fast.py "%s"' % (python_exec, language) cmd = '"%s" GPT_SoVITS/inference_webui_fast.py "%s"' % (python_exec, language)
else: else:
cmd = '"%s" GPT_SoVITS/inference_webui.py "%s"' % (python_exec, language) cmd = '"%s" GPT_SoVITS/inference_webui.py "%s"' % (python_exec, language)
#####v3暂不支持加速推理 # v3暂不支持加速推理
if version == "v3": if version == "v3":
cmd = '"%s" GPT_SoVITS/inference_webui.py "%s"' % (python_exec, language) cmd = '"%s" GPT_SoVITS/inference_webui.py "%s"' % (python_exec, language)
if p_tts_inference is None: if p_tts_inference is None:
@ -318,9 +373,10 @@ def change_tts_inference(bert_path,cnhubert_base_path,gpu_number,gpt_path,sovits
p_tts_inference = None p_tts_inference = None
yield process_info(process_name_tts, "closed"), {'__type__': 'update', 'visible': True}, {'__type__': 'update', 'visible': False} yield process_info(process_name_tts, "closed"), {'__type__': 'update', 'visible': True}, {'__type__': 'update', 'visible': False}
from tools.asr.config import asr_dict
process_name_asr = i18n("语音识别") process_name_asr = i18n("语音识别")
def open_asr(asr_inp_dir, asr_opt_dir, asr_model, asr_model_size, asr_lang, asr_precision): def open_asr(asr_inp_dir, asr_opt_dir, asr_model, asr_model_size, asr_lang, asr_precision):
global p_asr global p_asr
if p_asr is None: if p_asr is None:
@ -345,6 +401,7 @@ def open_asr(asr_inp_dir, asr_opt_dir, asr_model, asr_model_size, asr_lang, asr_
else: else:
yield process_info(process_name_asr, "occupy"), {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True}, {"__type__": "update"}, {"__type__": "update"}, {"__type__": "update"} yield process_info(process_name_asr, "occupy"), {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True}, {"__type__": "update"}, {"__type__": "update"}, {"__type__": "update"}
def close_asr(): def close_asr():
global p_asr global p_asr
if p_asr is not None: if p_asr is not None:
@ -352,7 +409,10 @@ def close_asr():
p_asr = None p_asr = None
return process_info(process_name_asr, "closed"), {"__type__": "update", "visible": True}, {"__type__": "update", "visible": False} return process_info(process_name_asr, "closed"), {"__type__": "update", "visible": True}, {"__type__": "update", "visible": False}
process_name_denoise = i18n("语音降噪") process_name_denoise = i18n("语音降噪")
def open_denoise(denoise_inp_dir, denoise_opt_dir): def open_denoise(denoise_inp_dir, denoise_opt_dir):
global p_denoise global p_denoise
if (p_denoise == None): if (p_denoise == None):
@ -370,6 +430,7 @@ def open_denoise(denoise_inp_dir, denoise_opt_dir):
else: else:
yield process_info(process_name_denoise, "occupy"), {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True}, {"__type__": "update"}, {"__type__": "update"} yield process_info(process_name_denoise, "occupy"), {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True}, {"__type__": "update"}, {"__type__": "update"}
def close_denoise(): def close_denoise():
global p_denoise global p_denoise
if p_denoise is not None: if p_denoise is not None:
@ -377,8 +438,11 @@ def close_denoise():
p_denoise = None p_denoise = None
return process_info(process_name_denoise, "closed"), {"__type__": "update", "visible": True}, {"__type__": "update", "visible": False} return process_info(process_name_denoise, "closed"), {"__type__": "update", "visible": True}, {"__type__": "update", "visible": False}
p_train_SoVITS = None p_train_SoVITS = None
process_name_sovits = i18n("SoVITS训练") process_name_sovits = i18n("SoVITS训练")
def open1Ba(batch_size, total_epoch, exp_name, text_low_lr_rate, if_save_latest, if_save_every_weights, save_every_epoch, gpu_numbers1Ba, pretrained_s2G, pretrained_s2D, if_grad_ckpt, lora_rank): def open1Ba(batch_size, total_epoch, exp_name, text_low_lr_rate, if_save_latest, if_save_every_weights, save_every_epoch, gpu_numbers1Ba, pretrained_s2G, pretrained_s2D, if_grad_ckpt, lora_rank):
global p_train_SoVITS global p_train_SoVITS
if (p_train_SoVITS == None): if (p_train_SoVITS == None):
@ -409,7 +473,8 @@ def open1Ba(batch_size,total_epoch,exp_name,text_low_lr_rate,if_save_latest,if_s
data["name"] = exp_name data["name"] = exp_name
data["version"] = version data["version"] = version
tmp_config_path = "%s/tmp_s2.json" % tmp tmp_config_path = "%s/tmp_s2.json" % tmp
with open(tmp_config_path,"w")as f:f.write(json.dumps(data)) with open(tmp_config_path, "w")as f:
f.write(json.dumps(data))
if version in ["v1", "v2"]: if version in ["v1", "v2"]:
cmd = '"%s" GPT_SoVITS/s2_train.py --config "%s"' % (python_exec, tmp_config_path) cmd = '"%s" GPT_SoVITS/s2_train.py --config "%s"' % (python_exec, tmp_config_path)
else: else:
@ -423,6 +488,7 @@ def open1Ba(batch_size,total_epoch,exp_name,text_low_lr_rate,if_save_latest,if_s
else: else:
yield process_info(process_name_sovits, "occupy"), {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True} yield process_info(process_name_sovits, "occupy"), {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True}
def close1Ba(): def close1Ba():
global p_train_SoVITS global p_train_SoVITS
if p_train_SoVITS is not None: if p_train_SoVITS is not None:
@ -430,8 +496,11 @@ def close1Ba():
p_train_SoVITS = None p_train_SoVITS = None
return process_info(process_name_sovits, "closed"), {"__type__": "update", "visible": True}, {"__type__": "update", "visible": False} return process_info(process_name_sovits, "closed"), {"__type__": "update", "visible": True}, {"__type__": "update", "visible": False}
p_train_GPT = None p_train_GPT = None
process_name_gpt = i18n("GPT训练") process_name_gpt = i18n("GPT训练")
def open1Bb(batch_size, total_epoch, exp_name, if_dpo, if_save_latest, if_save_every_weights, save_every_epoch, gpu_numbers, pretrained_s1): def open1Bb(batch_size, total_epoch, exp_name, if_dpo, if_save_latest, if_save_every_weights, save_every_epoch, gpu_numbers, pretrained_s1):
global p_train_GPT global p_train_GPT
if (p_train_GPT == None): if (p_train_GPT == None):
@ -462,7 +531,8 @@ def open1Bb(batch_size,total_epoch,exp_name,if_dpo,if_save_latest,if_save_every_
os.environ["_CUDA_VISIBLE_DEVICES"] = fix_gpu_numbers(gpu_numbers.replace("-", ",")) os.environ["_CUDA_VISIBLE_DEVICES"] = fix_gpu_numbers(gpu_numbers.replace("-", ","))
os.environ["hz"] = "25hz" os.environ["hz"] = "25hz"
tmp_config_path = "%s/tmp_s1.yaml" % tmp tmp_config_path = "%s/tmp_s1.yaml" % tmp
with open(tmp_config_path, "w") as f:f.write(yaml.dump(data, default_flow_style=False)) with open(tmp_config_path, "w") as f:
f.write(yaml.dump(data, default_flow_style=False))
# cmd = '"%s" GPT_SoVITS/s1_train.py --config_file "%s" --train_semantic_path "%s/6-name2semantic.tsv" --train_phoneme_path "%s/2-name2text.txt" --output_dir "%s/logs_s1"'%(python_exec,tmp_config_path,s1_dir,s1_dir,s1_dir) # cmd = '"%s" GPT_SoVITS/s1_train.py --config_file "%s" --train_semantic_path "%s/6-name2semantic.tsv" --train_phoneme_path "%s/2-name2text.txt" --output_dir "%s/logs_s1"'%(python_exec,tmp_config_path,s1_dir,s1_dir,s1_dir)
cmd = '"%s" GPT_SoVITS/s1_train.py --config_file "%s" ' % (python_exec, tmp_config_path) cmd = '"%s" GPT_SoVITS/s1_train.py --config_file "%s" ' % (python_exec, tmp_config_path)
yield process_info(process_name_gpt, "opened"), {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True} yield process_info(process_name_gpt, "opened"), {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True}
@ -474,6 +544,7 @@ def open1Bb(batch_size,total_epoch,exp_name,if_dpo,if_save_latest,if_save_every_
else: else:
yield process_info(process_name_gpt, "occupy"), {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True} yield process_info(process_name_gpt, "occupy"), {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True}
def close1Bb(): def close1Bb():
global p_train_GPT global p_train_GPT
if p_train_GPT is not None: if p_train_GPT is not None:
@ -481,8 +552,11 @@ def close1Bb():
p_train_GPT = None p_train_GPT = None
return process_info(process_name_gpt, "closed"), {"__type__": "update", "visible": True}, {"__type__": "update", "visible": False} return process_info(process_name_gpt, "closed"), {"__type__": "update", "visible": True}, {"__type__": "update", "visible": False}
ps_slice = [] ps_slice = []
process_name_slice = i18n("语音切分") process_name_slice = i18n("语音切分")
def open_slice(inp, opt_root, threshold, min_length, min_interval, hop_size, max_sil_kept, _max, alpha, n_parts): def open_slice(inp, opt_root, threshold, min_length, min_interval, hop_size, max_sil_kept, _max, alpha, n_parts):
global ps_slice global ps_slice
inp = my_utils.clean_path(inp) inp = my_utils.clean_path(inp)
@ -491,8 +565,10 @@ def open_slice(inp,opt_root,threshold,min_length,min_interval,hop_size,max_sil_k
if (os.path.exists(inp) == False): if (os.path.exists(inp) == False):
yield i18n("输入路径不存在"), {"__type__": "update", "visible": True}, {"__type__": "update", "visible": False}, {"__type__": "update"}, {"__type__": "update"}, {"__type__": "update"} yield i18n("输入路径不存在"), {"__type__": "update", "visible": True}, {"__type__": "update", "visible": False}, {"__type__": "update"}, {"__type__": "update"}, {"__type__": "update"}
return return
if os.path.isfile(inp):n_parts=1 if os.path.isfile(inp):
elif os.path.isdir(inp):pass n_parts = 1
elif os.path.isdir(inp):
pass
else: else:
yield i18n("输入路径存在但不可用"), {"__type__": "update", "visible": True}, {"__type__": "update", "visible": False}, {"__type__": "update"}, {"__type__": "update"}, {"__type__": "update"} yield i18n("输入路径存在但不可用"), {"__type__": "update", "visible": True}, {"__type__": "update", "visible": False}, {"__type__": "update"}, {"__type__": "update"}, {"__type__": "update"}
return return
@ -510,6 +586,7 @@ def open_slice(inp,opt_root,threshold,min_length,min_interval,hop_size,max_sil_k
else: else:
yield process_info(process_name_slice, "occupy"), {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True}, {"__type__": "update"}, {"__type__": "update"}, {"__type__": "update"} yield process_info(process_name_slice, "occupy"), {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True}, {"__type__": "update"}, {"__type__": "update"}, {"__type__": "update"}
def close_slice(): def close_slice():
global ps_slice global ps_slice
if (ps_slice != []): if (ps_slice != []):
@ -521,8 +598,11 @@ def close_slice():
ps_slice = [] ps_slice = []
return process_info(process_name_slice, "closed"), {"__type__": "update", "visible": True}, {"__type__": "update", "visible": False} return process_info(process_name_slice, "closed"), {"__type__": "update", "visible": True}, {"__type__": "update", "visible": False}
ps1a = [] ps1a = []
process_name_1a = i18n("文本分词与特征提取") process_name_1a = i18n("文本分词与特征提取")
def open1a(inp_text, inp_wav_dir, exp_name, gpu_numbers, bert_pretrained_dir): def open1a(inp_text, inp_wav_dir, exp_name, gpu_numbers, bert_pretrained_dir):
global ps1a global ps1a
inp_text = my_utils.clean_path(inp_text) inp_text = my_utils.clean_path(inp_text)
@ -574,6 +654,7 @@ def open1a(inp_text,inp_wav_dir,exp_name,gpu_numbers,bert_pretrained_dir):
else: else:
yield process_info(process_name_1a, "occupy"), {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True} yield process_info(process_name_1a, "occupy"), {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True}
def close1a(): def close1a():
global ps1a global ps1a
if ps1a != []: if ps1a != []:
@ -585,8 +666,11 @@ def close1a():
ps1a = [] ps1a = []
return process_info(process_name_1a, "closed"), {"__type__": "update", "visible": True}, {"__type__": "update", "visible": False} return process_info(process_name_1a, "closed"), {"__type__": "update", "visible": True}, {"__type__": "update", "visible": False}
ps1b = [] ps1b = []
process_name_1b = i18n("语音自监督特征提取") process_name_1b = i18n("语音自监督特征提取")
def open1b(inp_text, inp_wav_dir, exp_name, gpu_numbers, ssl_pretrained_dir): def open1b(inp_text, inp_wav_dir, exp_name, gpu_numbers, ssl_pretrained_dir):
global ps1b global ps1b
inp_text = my_utils.clean_path(inp_text) inp_text = my_utils.clean_path(inp_text)
@ -625,6 +709,7 @@ def open1b(inp_text,inp_wav_dir,exp_name,gpu_numbers,ssl_pretrained_dir):
else: else:
yield process_info(process_name_1b, "occupy"), {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True} yield process_info(process_name_1b, "occupy"), {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True}
def close1b(): def close1b():
global ps1b global ps1b
if (ps1b != []): if (ps1b != []):
@ -636,8 +721,11 @@ def close1b():
ps1b = [] ps1b = []
return process_info(process_name_1b, "closed"), {"__type__": "update", "visible": True}, {"__type__": "update", "visible": False} return process_info(process_name_1b, "closed"), {"__type__": "update", "visible": True}, {"__type__": "update", "visible": False}
ps1c = [] ps1c = []
process_name_1c = i18n("语义Token提取") process_name_1c = i18n("语义Token提取")
def open1c(inp_text, exp_name, gpu_numbers, pretrained_s2G_path): def open1c(inp_text, exp_name, gpu_numbers, pretrained_s2G_path):
global ps1c global ps1c
inp_text = my_utils.clean_path(inp_text) inp_text = my_utils.clean_path(inp_text)
@ -685,6 +773,7 @@ def open1c(inp_text,exp_name,gpu_numbers,pretrained_s2G_path):
else: else:
yield process_info(process_name_1c, "occupy"), {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True} yield process_info(process_name_1c, "occupy"), {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True}
def close1c(): def close1c():
global ps1c global ps1c
if (ps1c != []): if (ps1c != []):
@ -696,8 +785,11 @@ def close1c():
ps1c = [] ps1c = []
return process_info(process_name_1c, "closed"), {"__type__": "update", "visible": True}, {"__type__": "update", "visible": False} return process_info(process_name_1c, "closed"), {"__type__": "update", "visible": True}, {"__type__": "update", "visible": False}
ps1abc = [] ps1abc = []
process_name_1abc = i18n("训练集格式化一键三连") process_name_1abc = i18n("训练集格式化一键三连")
def open1abc(inp_text, inp_wav_dir, exp_name, gpu_numbers1a, gpu_numbers1Ba, gpu_numbers1c, bert_pretrained_dir, ssl_pretrained_dir, pretrained_s2G_path): def open1abc(inp_text, inp_wav_dir, exp_name, gpu_numbers1a, gpu_numbers1Ba, gpu_numbers1c, bert_pretrained_dir, ssl_pretrained_dir, pretrained_s2G_path):
global ps1abc global ps1abc
inp_text = my_utils.clean_path(inp_text) inp_text = my_utils.clean_path(inp_text)
@ -707,7 +799,7 @@ def open1abc(inp_text,inp_wav_dir,exp_name,gpu_numbers1a,gpu_numbers1Ba,gpu_numb
if (ps1abc == []): if (ps1abc == []):
opt_dir = "%s/%s" % (exp_root, exp_name) opt_dir = "%s/%s" % (exp_root, exp_name)
try: try:
#############################1a # 1a
path_text = "%s/2-name2text.txt" % opt_dir path_text = "%s/2-name2text.txt" % opt_dir
if (os.path.exists(path_text) == False or (os.path.exists(path_text) == True and len(open(path_text, "r", encoding="utf8").read().strip("\n").split("\n")) < 2)): if (os.path.exists(path_text) == False or (os.path.exists(path_text) == True and len(open(path_text, "r", encoding="utf8").read().strip("\n").split("\n")) < 2)):
config = { config = {
@ -734,7 +826,8 @@ def open1abc(inp_text,inp_wav_dir,exp_name,gpu_numbers1a,gpu_numbers1Ba,gpu_numb
p = Popen(cmd, shell=True) p = Popen(cmd, shell=True)
ps1abc.append(p) ps1abc.append(p)
yield i18n("进度") + ": 1A-Doing", {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True} yield i18n("进度") + ": 1A-Doing", {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True}
for p in ps1abc:p.wait() for p in ps1abc:
p.wait()
opt = [] opt = []
for i_part in range(all_parts): # txt_path="%s/2-name2text-%s.txt"%(opt_dir,i_part) for i_part in range(all_parts): # txt_path="%s/2-name2text-%s.txt"%(opt_dir,i_part)
@ -747,7 +840,7 @@ def open1abc(inp_text,inp_wav_dir,exp_name,gpu_numbers1a,gpu_numbers1Ba,gpu_numb
assert len("".join(opt)) > 0, process_info(process_name_1a, "failed") assert len("".join(opt)) > 0, process_info(process_name_1a, "failed")
yield i18n("进度") + ": 1A-Done", {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True} yield i18n("进度") + ": 1A-Done", {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True}
ps1abc = [] ps1abc = []
#############################1b # 1b
config = { config = {
"inp_text": inp_text, "inp_text": inp_text,
"inp_wav_dir": inp_wav_dir, "inp_wav_dir": inp_wav_dir,
@ -771,10 +864,11 @@ def open1abc(inp_text,inp_wav_dir,exp_name,gpu_numbers1a,gpu_numbers1Ba,gpu_numb
p = Popen(cmd, shell=True) p = Popen(cmd, shell=True)
ps1abc.append(p) ps1abc.append(p)
yield i18n("进度") + ": 1A-Done, 1B-Doing", {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True} yield i18n("进度") + ": 1A-Done, 1B-Doing", {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True}
for p in ps1abc:p.wait() for p in ps1abc:
p.wait()
yield i18n("进度") + ": 1A-Done, 1B-Done", {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True} yield i18n("进度") + ": 1A-Done, 1B-Done", {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True}
ps1abc = [] ps1abc = []
#############################1c # 1c
path_semantic = "%s/6-name2semantic.tsv" % opt_dir path_semantic = "%s/6-name2semantic.tsv" % opt_dir
if (os.path.exists(path_semantic) == False or (os.path.exists(path_semantic) == True and os.path.getsize(path_semantic) < 31)): if (os.path.exists(path_semantic) == False or (os.path.exists(path_semantic) == True and os.path.getsize(path_semantic) < 31)):
config = { config = {
@ -800,7 +894,8 @@ def open1abc(inp_text,inp_wav_dir,exp_name,gpu_numbers1a,gpu_numbers1Ba,gpu_numb
p = Popen(cmd, shell=True) p = Popen(cmd, shell=True)
ps1abc.append(p) ps1abc.append(p)
yield i18n("进度") + ": 1A-Done, 1B-Done, 1C-Doing", {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True} yield i18n("进度") + ": 1A-Done, 1B-Done, 1C-Doing", {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True}
for p in ps1abc:p.wait() for p in ps1abc:
p.wait()
opt = ["item_name\tsemantic_audio"] opt = ["item_name\tsemantic_audio"]
for i_part in range(all_parts): for i_part in range(all_parts):
@ -820,6 +915,7 @@ def open1abc(inp_text,inp_wav_dir,exp_name,gpu_numbers1a,gpu_numbers1Ba,gpu_numb
else: else:
yield process_info(process_name_1abc, "occupy"), {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True} yield process_info(process_name_1abc, "occupy"), {"__type__": "update", "visible": False}, {"__type__": "update", "visible": True}
def close1abc(): def close1abc():
global ps1abc global ps1abc
if (ps1abc != []): if (ps1abc != []):
@ -831,11 +927,13 @@ def close1abc():
ps1abc = [] ps1abc = []
return process_info(process_name_1abc, "closed"), {"__type__": "update", "visible": True}, {"__type__": "update", "visible": False} return process_info(process_name_1abc, "closed"), {"__type__": "update", "visible": True}, {"__type__": "update", "visible": False}
def switch_version(version_): def switch_version(version_):
os.environ["version"] = version_ os.environ["version"] = version_
global version global version
version = version_ version = version_
if pretrained_sovits_name[int(version[-1])-1] !='' and pretrained_gpt_name[int(version[-1])-1] !='':... if pretrained_sovits_name[int(version[-1]) - 1] != '' and pretrained_gpt_name[int(version[-1]) - 1] != '':
...
else: else:
gr.Warning(i18n('未下载模型') + ": " + version.upper()) gr.Warning(i18n('未下载模型') + ": " + version.upper())
set_default() set_default()
@ -852,23 +950,25 @@ def switch_version(version_):
{'__type__': 'update', "interactive": False if version == "v3" else True, "value": False}, \ {'__type__': 'update', "interactive": False if version == "v3" else True, "value": False}, \
{'__type__': 'update', "visible": True if version == "v3" else False} {'__type__': 'update', "visible": True if version == "v3" else False}
if os.path.exists('GPT_SoVITS/text/G2PWModel'):...
if os.path.exists('GPT_SoVITS/text/G2PWModel'):
...
else: else:
cmd = '"%s" GPT_SoVITS/download.py' % python_exec cmd = '"%s" GPT_SoVITS/download.py' % python_exec
p = Popen(cmd, shell=True) p = Popen(cmd, shell=True)
p.wait() p.wait()
def sync(text): def sync(text):
return {'__type__': 'update', 'value': text} return {'__type__': 'update', 'value': text}
with gr.Blocks(title="GPT-SoVITS WebUI") as app: with gr.Blocks(title="GPT-SoVITS WebUI") as app:
gr.Markdown( gr.Markdown(
value= value=i18n("本软件以MIT协议开源, 作者不对软件具备任何控制力, 使用软件者、传播软件导出的声音者自负全责.") + "<br>" + i18n("如不认可该条款, 则不能使用或引用软件包内任何代码和文件. 详见根目录LICENSE.")
i18n("本软件以MIT协议开源, 作者不对软件具备任何控制力, 使用软件者、传播软件导出的声音者自负全责.") + "<br>" + i18n("如不认可该条款, 则不能使用或引用软件包内任何代码和文件. 详见根目录LICENSE.")
) )
gr.Markdown( gr.Markdown(
value= value=i18n("中文教程文档") + ": " + "https://www.yuque.com/baicaigongchang1145haoyuangong/ib3g1e"
i18n("中文教程文档") + ": " + "https://www.yuque.com/baicaigongchang1145haoyuangong/ib3g1e"
) )
with gr.Tabs(): with gr.Tabs():
@ -931,8 +1031,10 @@ with gr.Blocks(title="GPT-SoVITS WebUI") as app:
def change_lang_choices(key): # 根据选择的模型修改可选的语言 def change_lang_choices(key): # 根据选择的模型修改可选的语言
return {"__type__": "update", "choices": asr_dict[key]['lang'], "value": asr_dict[key]['lang'][0]} return {"__type__": "update", "choices": asr_dict[key]['lang'], "value": asr_dict[key]['lang'][0]}
def change_size_choices(key): # 根据选择的模型修改可选的模型尺寸 def change_size_choices(key): # 根据选择的模型修改可选的模型尺寸
return {"__type__": "update", "choices": asr_dict[key]['size'], "value": asr_dict[key]['size'][-1]} return {"__type__": "update", "choices": asr_dict[key]['size'], "value": asr_dict[key]['size'][-1]}
def change_precision_choices(key): # 根据选择的模型修改可选的语言 def change_precision_choices(key): # 根据选择的模型修改可选的语言
if key == "Faster Whisper (多语种)": if key == "Faster Whisper (多语种)":
if default_batch_size <= 4: if default_batch_size <= 4:
@ -1121,7 +1223,8 @@ with gr.Blocks(title="GPT-SoVITS WebUI") as app:
version_checkbox.change(switch_version, [version_checkbox], [pretrained_s2G, pretrained_s2D, pretrained_s1, GPT_dropdown, SoVITS_dropdown, batch_size, total_epoch, save_every_epoch, text_low_lr_rate, if_grad_ckpt, batched_infer_enabled, lora_rank]) version_checkbox.change(switch_version, [version_checkbox], [pretrained_s2G, pretrained_s2D, pretrained_s1, GPT_dropdown, SoVITS_dropdown, batch_size, total_epoch, save_every_epoch, text_low_lr_rate, if_grad_ckpt, batched_infer_enabled, lora_rank])
with gr.TabItem(i18n("2-GPT-SoVITS-变声")):gr.Markdown(value=i18n("施工中，请静候佳音")) with gr.TabItem(i18n("2-GPT-SoVITS-变声")):
gr.Markdown(value=i18n("施工中，请静候佳音"))
app.queue().launch( # concurrency_count=511, max_size=1022 app.queue().launch( # concurrency_count=511, max_size=1022
server_name="0.0.0.0", server_name="0.0.0.0",