From 57843a21115ff9b06d40d22ed9f2ddfae3c9a2ba Mon Sep 17 00:00:00 2001 From: Feiteng Date: Fri, 9 May 2025 07:36:53 +0000 Subject: [PATCH] Fix msk frame_num --- wan/image2video.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/wan/image2video.py b/wan/image2video.py index 5004f46..bc73eb5 100644 --- a/wan/image2video.py +++ b/wan/image2video.py @@ -204,7 +204,7 @@ class WanI2V: generator=seed_g, device=self.device) - msk = torch.ones(1, 81, lat_h, lat_w, device=self.device) + msk = torch.ones(1, frame_num, lat_h, lat_w, device=self.device) msk[:, 1:] = 0 msk = torch.concat([ torch.repeat_interleave(msk[:, 0:1], repeats=4, dim=1), msk[:, 1:]