AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

[Wan] Fix mask padding in Wan VACE pipeline. (#11778)

Browse Source
This commit is contained in:
Yuanchen Guo
2025-06-23 18:58:21 +08:00
committed by GitHub
parent cd813499be
commit 798265f2b6
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/pipelines/wan/pipeline_wan_vace.py
View File

@@ -593,7 +593,7 @@ class WanVACEPipeline(DiffusionPipeline, WanLoraLoaderMixin):
num_ref_images = len(reference_images_batch)
if num_ref_images > 0:
mask_padding = torch.zeros_like(mask_[:, :num_ref_images, :, :])
mask_ = torch.cat([mask_, mask_padding], dim=1)
mask_ = torch.cat([mask_padding, mask_], dim=1)
mask_list.append(mask_)
return torch.stack(mask_list)