From a10986ad3009de40e71b9b51d62d3fbd9aa44a8a Mon Sep 17 00:00:00 2001 From: bmaltais Date: Tue, 31 Dec 2024 09:46:42 -0500 Subject: [PATCH] Fix network_module issue --- kohya_gui/lora_gui.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/kohya_gui/lora_gui.py b/kohya_gui/lora_gui.py index 054bc005..ce87cc3e 100644 --- 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()