From 0f7f457b82f3cfadc46aaca5bc8b7aca7135204d Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Fri, 20 Sep 2024 11:48:20 +0300
Subject: [PATCH] Don't keep VAE tiling enabled after decoding is done

---
 nodes.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/nodes.py b/nodes.py
index 42f7013..d4000d2 100644
--- 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()