VAE Tiling Support - credits to deepbeepmeep's WanGP

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

View File

@ -345,7 +345,9 @@ class WanI2V:
torch.cuda.empty_cache()
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 sample_scheduler