From 248428ccf849bd56cc3f91981c13b38b045f5352 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Fri, 30 Aug 2024 20:14:39 +0300
Subject: [PATCH] don't keep vae tiling enabled forever

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

diff --git a/nodes.py b/nodes.py
index dd80567..e97237d 100644
--- a/nodes.py
+++ b/nodes.py
@@ -353,6 +353,8 @@ class CogVideoDecode:
                 tile_overlap_factor_height=tile_overlap_factor_height,
                 tile_overlap_factor_width=tile_overlap_factor_width,
             )
+        else:
+            vae.disable_tiling()
         latents = latents.to(vae.dtype)
         latents = latents.permute(0, 2, 1, 3, 4)  # [batch_size, num_channels, num_frames, height, width]
         latents = 1 / vae.config.scaling_factor * latents