Fix network_module issue

bmaltais · Dec 31, 2024 · a10986a · a10986a
1 parent 8afcaf8
commit a10986a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kohya_gui/lora_gui.py b/kohya_gui/lora_gui.py
@@ -1342,7 +1342,7 @@ def train_model(
             "rank_dropout",
             "module_dropout",
         ]
-        network_module = "networks.lora" if sd3_checkbox else "networks.lora"
+        network_module = "networks.lora_sd3" if sd3_checkbox else "networks.lora"
         kohya_lora_vars = {
             key: value
             for key, value in vars().items()