From 841fe5237bbf8724e08870b1e83213e669cda0d1 Mon Sep 17 00:00:00 2001
From: Shiguang Ai <67623446+aishiguang@users.noreply.github.com>
Date: Tue, 2 Dec 2025 14:53:33 +0800
Subject: [PATCH] When frame_num is changed from 81 the tensor does not match
 (#533)

---
 wan/image2video.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/wan/image2video.py b/wan/image2video.py
index 6882c53..9aea726 100644
--- a/wan/image2video.py
+++ b/wan/image2video.py
@@ -207,7 +207,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, F, 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:]