Skip to content

Commit 0c0637f

Browse files
chore: update util function name as convention
1 parent 6efad43 commit 0c0637f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

invokeai/backend/patches/lora_conversions/flux_diffusers_lora_conversion_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ def add_lora_layer_if_present(src_key: str, dst_key: str) -> None:
8282
values = get_lora_layer_values(src_layer_dict)
8383
layers[dst_key] = any_lora_layer_from_state_dict(values)
8484

85-
def add_lora_adaLN_layer_if_present(src_key: str, dst_key: str) -> None:
85+
def add_adaLN_lora_layer_if_present(src_key: str, dst_key: str) -> None:
8686
if src_key in grouped_state_dict:
8787
src_layer_dict = grouped_state_dict.pop(src_key)
8888
values = get_lora_layer_values(src_layer_dict)
@@ -236,7 +236,7 @@ def add_qkv_lora_layer_if_present(
236236

237237
# Final layer.
238238
add_lora_layer_if_present("proj_out", "final_layer.linear")
239-
add_lora_adaLN_layer_if_present(
239+
add_adaLN_lora_layer_if_present(
240240
'norm_out.linear',
241241
'final_layer.adaLN_modulation.1',
242242
)

0 commit comments

Comments
 (0)