Fix max_train_steps_info error

bmaltais · Nov 18, 2024 · 309a9bb · 309a9bb
1 parent a6f0ff7
commit 309a9bb
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 0 deletions.
diff --git a/kohya_gui/dreambooth_gui.py b/kohya_gui/dreambooth_gui.py
@@ -654,6 +654,8 @@ def train_model(
         gr.Button(visible=False or headless),
         gr.Textbox(value=train_state_value),
     ]
+
+    max_train_steps_info = "Automatic by sd-scripts"
 
     if executor.is_running():
         log.error("Training is already running. Can't start another training session.")

diff --git a/kohya_gui/finetune_gui.py b/kohya_gui/finetune_gui.py
@@ -685,6 +685,8 @@ def train_model(
         gr.Button(visible=False or headless),
         gr.Textbox(value=train_state_value),
     ]
+
+    max_train_steps_info = "Automatic by sd-scripts"
 
     if executor.is_running():
         log.error("Training is already running. Can't start another training session.")

diff --git a/kohya_gui/lora_gui.py b/kohya_gui/lora_gui.py
@@ -883,6 +883,8 @@ def train_model(
         gr.Button(visible=False or headless),
         gr.Textbox(value=train_state_value),
     ]
+
+    max_train_steps_info = "Automatic by sd-scripts"
 
     if executor.is_running():
         log.error("Training is already running. Can't start another training session.")

diff --git a/kohya_gui/textual_inversion_gui.py b/kohya_gui/textual_inversion_gui.py
@@ -521,6 +521,8 @@ def train_model(
         gr.Textbox(value=train_state_value),
     ]
 
+    max_train_steps_info = "Automatic by sd-scripts"
+
     if executor.is_running():
         log.error("Training is already running. Can't start another training session.")
         return TRAIN_BUTTON_VISIBLE