Don't keep VAE tiling enabled after decoding is done

2025-12-28 12:54:03 +08:00 · 2024-09-20 11:48:20 +03:00 · 2024-09-20 11:48:20 +03:00 · 0f7f457b82
commit 0f7f457b82
parent a5908e556f
1 changed files with 1 additions and 0 deletions
--- a/nodes.py
+++ b/nodes.py
@ -523,6 +523,7 @@ class CogVideoDecode:
        latents = 1 / vae.config.scaling_factor * latents
       
        frames = vae.decode(latents).sample
+        vae.disable_tiling()
        if not pipeline["cpu_offloading"]:
            vae.to(offload_device)
        mm.soft_empty_cache()