From 7604f36bb270d0a897df0b8d4dd9d35f860d06cb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=80=B8=E6=B8=B8=E4=BB=99=E4=BA=BA?= <3439248121@qq.com>
Date: Sun, 22 Jun 2025 03:51:59 +0800
Subject: [PATCH] =?UTF-8?q?=E7=94=A8=E4=BA=8E=E7=89=B9=E6=AE=8A=E7=94=A8?=
 =?UTF-8?q?=E9=80=94=E7=9A=84=E5=A4=A7=E6=95=B0=E6=8D=AE=E9=87=8F=E5=AE=9E?=
 =?UTF-8?q?=E9=AA=8C=EF=BC=8C=E8=AF=B7=E5=8B=BF=E5=90=88=E5=B9=B6=EF=BC=81?=
 =?UTF-8?q?=EF=BC=81=EF=BC=81=EF=BC=81=EF=BC=81=EF=BC=81=EF=BC=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 webui.py | 23 +++++++++++++----------
 1 file changed, 13 insertions(+), 10 deletions(-)

diff --git a/webui.py b/webui.py
index 9981cfcc..c6060920 100644
--- a/webui.py
+++ b/webui.py
@@ -1,3 +1,6 @@
+import os
+os.environ["CUDA_VISIBLE_DEVICES"] = "0"  # 限制为单卡
+
 import os
 import sys
 
@@ -127,13 +130,13 @@ def set_default():
     if version not in v3v4set:
         default_sovits_epoch = 8
         default_sovits_save_every_epoch = 4
-        max_sovits_epoch = 25  # 40
-        max_sovits_save_every_epoch = 25  # 10
+        max_sovits_epoch = 255  # 40
+        max_sovits_save_every_epoch = 255  # 10
     else:
         default_sovits_epoch = 2
         default_sovits_save_every_epoch = 1
-        max_sovits_epoch = 16  # 40 # 3 #训太多=作死
-        max_sovits_save_every_epoch = 10  # 10 # 3
+        max_sovits_epoch = 255  # 40 # 3 #训太多=作死
+        max_sovits_save_every_epoch = 255  # 10 # 3
 
     default_batch_size = max(1, default_batch_size)
     default_batch_size_s1 = max(1, default_batch_size_s1)
@@ -507,7 +510,7 @@ def open1Ba(
 ):
     global p_train_SoVITS
     if p_train_SoVITS == None:
-        exp_name = exp_name.rstrip(" ")
+        exp_name=exp_name.rstrip(" ")
         config_file = (
             "GPT_SoVITS/configs/s2.json"
             if version not in {"v2Pro", "v2ProPlus"}
@@ -604,7 +607,7 @@ def open1Bb(
 ):
     global p_train_GPT
     if p_train_GPT == None:
-        exp_name = exp_name.rstrip(" ")
+        exp_name=exp_name.rstrip(" ")
         with open(
             "GPT_SoVITS/configs/s1longer.yaml" if version == "v1" else "GPT_SoVITS/configs/s1longer-v2.yaml"
         ) as f:
@@ -1728,8 +1731,8 @@ with gr.Blocks(title="GPT-SoVITS WebUI", analytics_enabled=False, js=js, css=css
                                 )
                             with gr.Row():
                                 text_low_lr_rate = gr.Slider(
-                                    minimum=0.2,
-                                    maximum=0.6,
+                                    minimum=0,
+                                    maximum=1,
                                     step=0.05,
                                     label=i18n("文本模块学习率权重"),
                                     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(
                                     label=i18n("LoRA秩"),
                                     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,
                                 )  # v1v2 not need
                                 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(
                                     minimum=2,
-                                    maximum=50,
+                                    maximum=max_sovits_epoch,
                                     step=1,
                                     label=i18n("总训练轮数total_epoch"),
                                     value=15,