fix

linoytsaban · linoytsaban · commit 6d62f4b47fe5 · 2025-05-20T16:05:28.000+03:00
diff --git a/src/diffusers/loaders/lora_pipeline.py b/src/diffusers/loaders/lora_pipeline.py
@@ -4833,7 +4833,7 @@ def _maybe_expand_t2v_lora_for_i2v(
                         continue
 
                     state_dict[f"transformer.blocks.{i}.attn2.{c}.lora_A.weight"] = torch.zeros_like(
-                        ref_lora_A_weight.shape, device=target_device
+                        state_dict[f"transformer.blocks.{i}.attn2.to_k.lora_A.weight"], device=target_device
                     )
                     state_dict[f"transformer.blocks.{i}.attn2.{c}.lora_B.weight"] = torch.zeros_like(
                         state_dict[f"transformer.blocks.{i}.attn2.to_k.lora_B.weight"], device=target_device

Original file line number	Diff line number	Diff line change
`@@ -4833,7 +4833,7 @@ def _maybe_expand_t2v_lora_for_i2v(`
`4833`	`4833`	`continue`
`4834`	`4834`
`4835`	`4835`	`state_dict[f"transformer.blocks.{i}.attn2.{c}.lora_A.weight"] = torch.zeros_like(`
`4836`		`- ref_lora_A_weight.shape, device=target_device`
	`4836`	`+ state_dict[f"transformer.blocks.{i}.attn2.to_k.lora_A.weight"], device=target_device`
`4837`	`4837`	`)`
`4838`	`4838`	`state_dict[f"transformer.blocks.{i}.attn2.{c}.lora_B.weight"] = torch.zeros_like(`
`4839`	`4839`	`state_dict[f"transformer.blocks.{i}.attn2.to_k.lora_B.weight"], device=target_device`