Skip to content

Commit 78a8e1e

Browse files
committed
vace padding
1 parent 5901c25 commit 78a8e1e

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

src/diffusers/loaders/lora_pipeline.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4810,12 +4810,11 @@ def _maybe_expand_t2v_lora_for_i2v(
48104810
transformer: torch.nn.Module,
48114811
state_dict,
48124812
):
4813-
if transformer.config.image_dim is None:
4814-
return state_dict
4813+
print("wtf 0", hasattr(transformer, 'vace_blocks'))
4814+
# if transformer.config.image_dim is None:
4815+
# return state_dict
48154816

48164817
target_device = transformer.device
4817-
print("expanding t2v to i2v")
4818-
print("wtf 0",hasattr(transformer, 'vace_blocks'))
48194818
if any(k.startswith("transformer.blocks.") for k in state_dict):
48204819
num_blocks = len({k.split("blocks.")[1].split(".")[0] for k in state_dict if "blocks." in k})
48214820
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)

0 commit comments

Comments
 (0)