Fix for max_train_steps

bmaltais · Sep 13, 2024 · 46b57ed · 46b57ed
1 parent 0d53ac6
commit 46b57ed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kohya_gui/lora_gui.py b/kohya_gui/lora_gui.py
@@ -991,7 +991,7 @@ def train_model(
         log.info(f"Total steps: {total_steps}")
 
     # Calculate stop_text_encoder_training
-    if max_train_steps > 0 and stop_text_encoder_training > 0:
+    if max_train_steps is not None and max_train_steps > 0 and stop_text_encoder_training > 0:
         stop_text_encoder_training = math.ceil(
             max_train_steps * stop_text_encoder_training / 100
         )