From 309a9bbc1b333d965592efc1a4c762b2bd489c1c Mon Sep 17 00:00:00 2001 From: bmaltais Date: Mon, 18 Nov 2024 07:07:33 -0500 Subject: [PATCH] Fix max_train_steps_info error --- kohya_gui/dreambooth_gui.py | 2 ++ kohya_gui/finetune_gui.py | 2 ++ kohya_gui/lora_gui.py | 2 ++ kohya_gui/textual_inversion_gui.py | 2 ++ 4 files changed, 8 insertions(+) diff --git a/kohya_gui/dreambooth_gui.py b/kohya_gui/dreambooth_gui.py index 98d67530..81db18ce 100644 --- 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 index e01456ca..d6f13710 100644 --- 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 index 75dbb423..53d7bb58 100644 --- 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 index 9b8ea6b2..618a0d81 100644 --- 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