Update lora_gui.py

Pent · Mar 19, 2024 · c3239a0 · c3239a0
1 parent 726f5b3
commit c3239a0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kohya_gui/lora_gui.py b/kohya_gui/lora_gui.py
@@ -719,7 +719,7 @@ def train_model(
 
     if LoRA_type == "LyCORIS/Diag-OFT":
         network_module = "lycoris.kohya"
-        network_args = f' "preset={LyCORIS_preset}" "conv_dim={conv_dim}" "conv_alpha={conv_alpha}" "module_dropout={module_dropout}" "use_tucker={use_tucker}" "use_scalar={use_scalar}" "rank_dropout_scale={rank_dropout_scale}" "constrain={constrain}" "rescaled={rescaled}" "algo=diag-oft" '
+        network_args = f' "preset={LyCORIS_preset}" "conv_dim={conv_dim}" "conv_alpha={conv_alpha}" "module_dropout={module_dropout}" "use_tucker={use_tucker}" "use_scalar={use_scalar}" "rank_dropout_scale={rank_dropout_scale}" "constrain={constrain}" "rescaled={rescaled}" "algo=diag-oft"  "train_norm={train_norm}"'
 
     if LoRA_type == "LyCORIS/DyLoRA":
         network_module = "lycoris.kohya"