From 8c11a08eb334437a931edb55cb96a0802b8dcf64 Mon Sep 17 00:00:00 2001 From: bmaltais Date: Mon, 16 Sep 2024 18:51:09 -0400 Subject: [PATCH] Add missing network parms --- kohya_gui/lora_gui.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/kohya_gui/lora_gui.py b/kohya_gui/lora_gui.py index ecfe495d..7978b6ca 100644 --- 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`