Allow for vae tiling onle without low vram on wan models with a model kwarg

2026-07-10 09:42:52 -06:00
parent fe82487187
commit 886c2aec57
3 changed files with 27 additions and 2 deletions
--- a/extensions_built_in/diffusion_models/wan22/wan22_5b_model.py
+++ b/extensions_built_in/diffusion_models/wan22/wan22_5b_model.py
@@ -210,7 +210,7 @@ class Wan225bModel(Wan21):
                latent_model_input=latents, first_frame=first_frame_n1p1, vae=self.vae
            )

-        if self.model_config.low_vram:
+        if self.use_vae_tiling:
            # set vae to tile decode
            pipeline.vae.enable_tiling()

@@ -234,7 +234,7 @@ class Wan225bModel(Wan21):
            **extra,
        )[0]

-        if self.model_config.low_vram:
+        if self.use_vae_tiling:
            # restore no tiling
            pipeline.vae.disable_tiling()