Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-06-06 23:34:53 +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 23:28:53 -07:00 committed by GitHub
parent 79f3915348
commit ae7320793c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
wan/image2video.py
View File

@ -345,7 +345,9 @@ class WanI2V:
torch.cuda.empty_cache() torch.cuda.empty_cache()
if self.rank == 0: if self.rank == 0:
videos = self.vae.decode(x0) videos = self.vae.decode(x0
, VAE_tile_size # 20250316 pftq: VAE tiling from deepbeepmeep/WanGP
)
del noise, latent del noise, latent
del sample_scheduler del sample_scheduler