Skip to content

Commit 51c570d

Browse files
linoytsabanhlky
andauthored
Update src/diffusers/loaders/lora_pipeline.py
Co-authored-by: hlky <[email protected]>
1 parent 92aabcb commit 51c570d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/loaders/lora_pipeline.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4257,7 +4257,7 @@ def _maybe_expand_t2v_lora_for_i2v(
42574257
):
42584258
if any(k.startswith("blocks.") for k in state_dict):
42594259
num_blocks = len({k.split("blocks.")[1].split(".")[0] for k in state_dict})
4260-
is_i2v_lora = any("k_img" in k for k in state_dict) and any("v_img" in k for k in state_dict)
4260+
is_i2v_lora = any("add_k_proj" in k for k in state_dict) and any("add_v_proj" in k for k in state_dict)
42614261
if is_i2v_lora:
42624262
return state_dict
42634263

0 commit comments

Comments
 (0)