Skip to content

Commit 309a9bb

Browse files
committed
Fix max_train_steps_info error
1 parent a6f0ff7 commit 309a9bb

File tree

4 files changed

+8
-0
lines changed

4 files changed

+8
-0
lines changed

kohya_gui/dreambooth_gui.py

+2
Original file line numberDiff line numberDiff line change
@@ -654,6 +654,8 @@ def train_model(
654654
gr.Button(visible=False or headless),
655655
gr.Textbox(value=train_state_value),
656656
]
657+
658+
max_train_steps_info = "Automatic by sd-scripts"
657659

658660
if executor.is_running():
659661
log.error("Training is already running. Can't start another training session.")

kohya_gui/finetune_gui.py

+2
Original file line numberDiff line numberDiff line change
@@ -685,6 +685,8 @@ def train_model(
685685
gr.Button(visible=False or headless),
686686
gr.Textbox(value=train_state_value),
687687
]
688+
689+
max_train_steps_info = "Automatic by sd-scripts"
688690

689691
if executor.is_running():
690692
log.error("Training is already running. Can't start another training session.")

kohya_gui/lora_gui.py

+2
Original file line numberDiff line numberDiff line change
@@ -883,6 +883,8 @@ def train_model(
883883
gr.Button(visible=False or headless),
884884
gr.Textbox(value=train_state_value),
885885
]
886+
887+
max_train_steps_info = "Automatic by sd-scripts"
886888

887889
if executor.is_running():
888890
log.error("Training is already running. Can't start another training session.")

kohya_gui/textual_inversion_gui.py

+2
Original file line numberDiff line numberDiff line change
@@ -521,6 +521,8 @@ def train_model(
521521
gr.Textbox(value=train_state_value),
522522
]
523523

524+
max_train_steps_info = "Automatic by sd-scripts"
525+
524526
if executor.is_running():
525527
log.error("Training is already running. Can't start another training session.")
526528
return TRAIN_BUTTON_VISIBLE

0 commit comments

Comments
 (0)