Skip to content

Commit 6bdfd15

Browse files
chore: update util function name as convention
1 parent 3876c4c commit 6bdfd15

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
@@ -99,7 +99,7 @@ def add_lora_layer_if_present(src_key: str, dst_key: str) -> None:
9999
values = get_lora_layer_values(src_layer_dict)
100100
layers[dst_key] = any_lora_layer_from_state_dict(values)
101101

102-
def add_lora_adaLN_layer_if_present(src_key: str, dst_key: str) -> None:
102+
def add_adaLN_lora_layer_if_present(src_key: str, dst_key: str) -> None:
103103
if src_key in grouped_state_dict:
104104
src_layer_dict = grouped_state_dict.pop(src_key)
105105
values = get_lora_layer_values(src_layer_dict)
@@ -253,7 +253,7 @@ def add_qkv_lora_layer_if_present(
253253

254254
# Final layer.
255255
add_lora_layer_if_present("proj_out", "final_layer.linear")
256-
add_lora_adaLN_layer_if_present(
256+
add_adaLN_lora_layer_if_present(
257257
'norm_out.linear',
258258
'final_layer.adaLN_modulation.1',
259259
)

0 commit comments

Comments
 (0)