1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00

vace padding

This commit is contained in:
linoytsaban
2025-05-21 18:24:05 +03:00
parent c0d22f8757
commit 46755bdf1a

View File

@@ -4814,7 +4814,8 @@ class WanLoraLoaderMixin(LoraBaseMixin):
return state_dict
target_device = transformer.device
print("expanding t2v to i2v")
print("wtf 0",hasattr(transformer, 'vace_blocks'))
if any(k.startswith("transformer.blocks.") for k in state_dict):
num_blocks = len({k.split("blocks.")[1].split(".")[0] for k in state_dict if "blocks." in k})
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)
@@ -4850,7 +4851,6 @@ class WanLoraLoaderMixin(LoraBaseMixin):
device=target_device,
)
print(hasattr(transformer, 'vace_blocks'))
if hasattr(transformer, 'vace_blocks'):
print(f"{i}, WTF 0")