From 4bae4fc2aa141e0ed445601dde0ae65ba35d248b Mon Sep 17 00:00:00 2001
From: bmaltais <bernard@ducourier.com>
Date: Sun, 8 Sep 2024 13:20:29 -0400
Subject: [PATCH] Validate if lora type is Flux1 when flux1_checkbox is true

---
 kohya_gui/lora_gui.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/kohya_gui/lora_gui.py b/kohya_gui/lora_gui.py
index be85de68..9b98375e 100644
--- 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
     #