Validate if lora type is Flux1 when flux1_checkbox is true

bmaltais · Sep 8, 2024 · 4bae4fc · 4bae4fc
1 parent 8f6229f
commit 4bae4fc
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/kohya_gui/lora_gui.py b/kohya_gui/lora_gui.py
@@ -836,6 +836,12 @@ def train_model(
     if not validate_args_setting(optimizer_args):
         return TRAIN_BUTTON_VISIBLE
 
+    if flux1_checkbox:
+        log.info(f"Validating lora type is Flux1 if flux1 checkbox is checked...")
+        if LoRA_type != "Flux1":
+            log.error("LoRA type must be set to Flux1 if Flux1 checkbox is checked.")
+            return TRAIN_BUTTON_VISIBLE
+
     #
     # Validate paths
     #