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

VAE Tiling Support - credits to deepbeepmeep's WanGP

Browse Source
This commit is contained in:
pftq 2025-03-18 22:17:31 -07:00 committed by GitHub
parent 09ba994635
commit d0ff85adc9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
wan/image2video.py
View File

@ -137,7 +137,9 @@ class WanI2V:
guide_scale=5.0, guide_scale=5.0,
n_prompt="", n_prompt="",
seed=-1, seed=-1,
offload_model=True): offload_model=True,
VAE_tile_size=0, # 20250316 pftq: vae tiling from deepbeepmeep/WanGP
):
r""" r"""
Generates video frames from input image and text prompt using diffusion process. Generates video frames from input image and text prompt using diffusion process.
@ -249,7 +251,9 @@ class WanI2V:
torch.zeros(3, F-1, h, w) # 20250226 pftq: fixed 80 being hardcoded frame-1 torch.zeros(3, F-1, h, w) # 20250226 pftq: fixed 80 being hardcoded frame-1
], ],
dim=1).to(self.device) dim=1).to(self.device)
])[0] ]
, VAE_tile_size # 20250316 pftq: VAE tiling from deepbeepmeep/WanGP
)[0]
y = torch.concat([msk, y]) y = torch.concat([msk, y])
@contextmanager @contextmanager