From ae7320793c08228d0e245e7f40ec34b7e7e5e8d7 Mon Sep 17 00:00:00 2001 From: pftq Date: Tue, 18 Mar 2025 23:28:53 -0700 Subject: [PATCH] VAE Tiling Support - credits to deepbeepmeep's WanGP --- wan/image2video.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/wan/image2video.py b/wan/image2video.py index df0eeed..6b828f3 100644 --- a/wan/image2video.py +++ b/wan/image2video.py @@ -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