Add missing network parms

bmaltais · Sep 16, 2024 · 8c11a08 · 8c11a08
1 parent 416ef0e
commit 8c11a08
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/kohya_gui/lora_gui.py b/kohya_gui/lora_gui.py
@@ -1181,6 +1181,8 @@ def train_model(
             "txt_mod_dim",
             "single_mod_dim",
             "in_dims",
+            "train_double_block_indices",
+            "train_single_block_indices",
         ]
         network_module = "networks.lora_flux"
         kohya_lora_vars = {
@@ -1542,8 +1544,6 @@ def train_model(
         "mem_eff_save": mem_eff_save if flux1_checkbox else None,
         "apply_t5_attn_mask": apply_t5_attn_mask if flux1_checkbox else None,
         "cpu_offload_checkpointing": cpu_offload_checkpointing if flux1_checkbox else None,
-        "train_double_block_indices": train_double_block_indices if flux1_checkbox else None,
-        "train_single_block_indices": train_single_block_indices if flux1_checkbox else None,
     }
 
     # Given dictionary `config_toml_data`