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

fix bug with mask

This commit is contained in:
Aryan
2025-05-21 02:19:30 +02:00
parent 694dcf2cfd
commit 23f6bc1201

View File

@@ -392,6 +392,7 @@ class WanVACEPipeline(DiffusionPipeline, WanLoraLoaderMixin):
if mask is not None:
mask = self.video_processor.preprocess_video(mask, image_size[0], image_size[1])
mask = torch.clamp((mask + 1) / 2, min=0, max=1)
else:
mask = torch.ones_like(video)