Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-06-07 15:54:52 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fixed I2V Frames Hardcoded at 81 - updated frame_num to F for consistency

Browse Source
This commit is contained in:
pftq 2025-02-26 13:22:15 -08:00 committed by GitHub
parent f9267254b5
commit de2b78a69c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
wan/image2video.py
View File

@ -205,7 +205,7 @@ class WanI2V:
device=self.device) device=self.device)
#20250226 fix noise assuming frames hardcoded at 81 (21 latent frames) #20250226 fix noise assuming frames hardcoded at 81 (21 latent frames)
latent_frame_num = (frame_num - 1) // self.vae_stride[0] + 1 latent_frame_num = (F - 1) // self.vae_stride[0] + 1
noise = torch.randn(16, latent_frame_num, lat_h, lat_w, dtype=torch.float32, generator=seed_g, device=self.device) noise = torch.randn(16, latent_frame_num, lat_h, lat_w, dtype=torch.float32, generator=seed_g, device=self.device)
#msk = torch.ones(1, 81, lat_h, lat_w, device=self.device) #msk = torch.ones(1, 81, lat_h, lat_w, device=self.device)