blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-08-12 11:33:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

训完模自动刷新模型列表

Browse Source 
Automatically refresh the model list after training the model.
This commit is contained in:
RVC-Boss 2025-04-21 21:30:36 +08:00 committed by GitHub
parent 9d481da610
commit ec7ec370ef
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 43 additions and 43 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

86
webui.py
View File

@ -197,7 +197,7 @@ def set_default():
else:
default_sovits_epoch = 2
default_sovits_save_every_epoch = 1
max_sovits_epoch = 50 # 40 # 3
max_sovits_epoch = 20 # 40 # 3
max_sovits_save_every_epoch = 10 # 10 # 3
default_batch_size = max(1, default_batch_size)
@ -589,7 +589,6 @@ def close_denoise():
p_train_SoVITS = None
process_name_sovits = i18n("SoVITS训练")
def open1Ba(
batch_size,
total_epoch,
@ -642,22 +641,23 @@ def open1Ba(
yield (
process_info(process_name_sovits, "opened"),
{"__type__": "update", "visible": False},
{"__type__": "update", "visible": True},
{"__type__": "update", "visible": True},{"__type__": "update"},{"__type__": "update"}
)
print(cmd)
p_train_SoVITS = Popen(cmd, shell=True)
p_train_SoVITS.wait()
p_train_SoVITS = None
SoVITS_dropdown_update, GPT_dropdown_update = change_choices()
yield (
process_info(process_name_sovits, "finish"),
{"__type__": "update", "visible": True},
{"__type__": "update", "visible": False},
{"__type__": "update", "visible": False},SoVITS_dropdown_update,GPT_dropdown_update
)
else:
yield (
process_info(process_name_sovits, "occupy"),
{"__type__": "update", "visible": False},
{"__type__": "update", "visible": True},
{"__type__": "update", "visible": True},{"__type__": "update"},{"__type__": "update"}
)
@ -726,22 +726,23 @@ def open1Bb(
yield (
process_info(process_name_gpt, "opened"),
{"__type__": "update", "visible": False},
{"__type__": "update", "visible": True},
{"__type__": "update", "visible": True},{"__type__": "update"},{"__type__": "update"}
)
print(cmd)
p_train_GPT = Popen(cmd, shell=True)
p_train_GPT.wait()
p_train_GPT = None
SoVITS_dropdown_update, GPT_dropdown_update = change_choices()
yield (
process_info(process_name_gpt, "finish"),
{"__type__": "update", "visible": True},
{"__type__": "update", "visible": False},
{"__type__": "update", "visible": False},SoVITS_dropdown_update,GPT_dropdown_update
)
else:
yield (
process_info(process_name_gpt, "occupy"),
{"__type__": "update", "visible": False},
{"__type__": "update", "visible": True},
{"__type__": "update", "visible": True},{"__type__": "update"},{"__type__": "update"}
)
@ -1719,7 +1720,7 @@ with gr.Blocks(title="GPT-SoVITS WebUI") as app:
label=i18n("文本模块学习率权重"),
value=0.4,
visible=True if version not in v3v4set else False,
) # v3 not need
) # v3v4 not need
lora_rank = gr.Radio(
label=i18n("LoRA秩"),
value="32",
@ -1833,40 +1834,7 @@ with gr.Blocks(title="GPT-SoVITS WebUI") as app:
with gr.Row():
info1Bb = gr.Textbox(label=process_info(process_name_gpt, "info"))
button1Ba_open.click(
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,
],
[info1Ba, button1Ba_open, button1Ba_close],
)
button1Ba_close.click(close1Ba, [], [info1Ba, button1Ba_open, button1Ba_close])
button1Bb_open.click(
open1Bb,
[
batch_size1Bb,
total_epoch1Bb,
exp_name,
if_dpo,
if_save_latest1Bb,
if_save_every_weights1Bb,
save_every_epoch1Bb,
gpu_numbers1Bb,
pretrained_s1,
],
[info1Bb, button1Bb_open, button1Bb_close],
)
button1Bb_close.click(close1Bb, [], [info1Bb, button1Bb_open, button1Bb_close])
with gr.TabItem("1C-" + i18n("推理")):
@ -1931,7 +1899,39 @@ with gr.Blocks(title="GPT-SoVITS WebUI") as app:
],
[tts_info, open_tts, close_tts],
)
button1Ba_open.click(
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,
],
[info1Ba, button1Ba_open, button1Ba_close,SoVITS_dropdown,GPT_dropdown],
)
button1Bb_open.click(
open1Bb,
[
batch_size1Bb,
total_epoch1Bb,
exp_name,
if_dpo,
if_save_latest1Bb,
if_save_every_weights1Bb,
save_every_epoch1Bb,
gpu_numbers1Bb,
pretrained_s1,
],
[info1Bb, button1Bb_open, button1Bb_close,SoVITS_dropdown,GPT_dropdown],
)
version_checkbox.change(
switch_version,
[version_checkbox],