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

用于特殊用途的大数据量实验，请勿合并！！！！！！！

Browse Source
This commit is contained in:
逸游仙人 2025-06-22 03:51:59 +08:00
parent 6c88f1ea32
commit 7604f36bb2
1 changed files with 13 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
webui.py
View File

@ -1,3 +1,6 @@
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0" # 限制为单卡
import os import os
import sys import sys
@ -127,13 +130,13 @@ def set_default():
if version not in v3v4set: if version not in v3v4set:
default_sovits_epoch = 8 default_sovits_epoch = 8
default_sovits_save_every_epoch = 4 default_sovits_save_every_epoch = 4
max_sovits_epoch = 25 # 40 max_sovits_epoch = 255 # 40
max_sovits_save_every_epoch = 25 # 10 max_sovits_save_every_epoch = 255 # 10
else: else:
default_sovits_epoch = 2 default_sovits_epoch = 2
default_sovits_save_every_epoch = 1 default_sovits_save_every_epoch = 1
max_sovits_epoch = 16 # 40 # 3 #训太多=作死 max_sovits_epoch = 255 # 40 # 3 #训太多=作死
max_sovits_save_every_epoch = 10 # 10 # 3 max_sovits_save_every_epoch = 255 # 10 # 3
default_batch_size = max(1, default_batch_size) default_batch_size = max(1, default_batch_size)
default_batch_size_s1 = max(1, default_batch_size_s1) default_batch_size_s1 = max(1, default_batch_size_s1)
@ -507,7 +510,7 @@ def open1Ba(
): ):
global p_train_SoVITS global p_train_SoVITS
if p_train_SoVITS == None: if p_train_SoVITS == None:
exp_name = exp_name.rstrip(" ") exp_name=exp_name.rstrip(" ")
config_file = ( config_file = (
"GPT_SoVITS/configs/s2.json" "GPT_SoVITS/configs/s2.json"
if version not in {"v2Pro", "v2ProPlus"} if version not in {"v2Pro", "v2ProPlus"}
@ -604,7 +607,7 @@ def open1Bb(
): ):
global p_train_GPT global p_train_GPT
if p_train_GPT == None: if p_train_GPT == None:
exp_name = exp_name.rstrip(" ") exp_name=exp_name.rstrip(" ")
with open( with open(
"GPT_SoVITS/configs/s1longer.yaml" if version == "v1" else "GPT_SoVITS/configs/s1longer-v2.yaml" "GPT_SoVITS/configs/s1longer.yaml" if version == "v1" else "GPT_SoVITS/configs/s1longer-v2.yaml"
) as f: ) as f:
@ -1728,8 +1731,8 @@ with gr.Blocks(title="GPT-SoVITS WebUI", analytics_enabled=False, js=js, css=css
) )
with gr.Row(): with gr.Row():
text_low_lr_rate = gr.Slider( text_low_lr_rate = gr.Slider(
minimum=0.2, minimum=0,
maximum=0.6, maximum=1,
step=0.05, step=0.05,
label=i18n("文本模块学习率权重"), label=i18n("文本模块学习率权重"),
value=0.4, value=0.4,
@ -1738,7 +1741,7 @@ with gr.Blocks(title="GPT-SoVITS WebUI", analytics_enabled=False, js=js, css=css
lora_rank = gr.Radio( lora_rank = gr.Radio(
label=i18n("LoRA秩"), label=i18n("LoRA秩"),
value="32", value="32",
choices=["16", "32", "64", "128"], choices=["16", "32", "64", "128", "256", "512", "1024","2048", "4096"],
visible=True if version in v3v4set else False, visible=True if version in v3v4set else False,
) # v1v2 not need ) # v1v2 not need
save_every_epoch = gr.Slider( save_every_epoch = gr.Slider(
@ -1800,7 +1803,7 @@ with gr.Blocks(title="GPT-SoVITS WebUI", analytics_enabled=False, js=js, css=css
) )
total_epoch1Bb = gr.Slider( total_epoch1Bb = gr.Slider(
minimum=2, minimum=2,
maximum=50, maximum=max_sovits_epoch,
step=1, step=1,
label=i18n("总训练轮数total_epoch"), label=i18n("总训练轮数total_epoch"),
value=15, value=15,