From bbbb6e9514bd484f52cdd0c56ad194622b59957b Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Fri, 30 Aug 2024 22:13:26 +0300 Subject: [PATCH] Update nodes.py --- nodes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nodes.py b/nodes.py index e33c956..a1f1e3a 100644 --- a/nodes.py +++ b/nodes.py @@ -185,7 +185,7 @@ class CogVideoImageEncode: FUNCTION = "encode" CATEGORY = "CogVideoWrapper" - def encode(self, pipeline, image, chunk_size=16, enable_vae_slicing=True): + def encode(self, pipeline, image, chunk_size=8, enable_vae_slicing=True): device = mm.get_torch_device() offload_device = mm.unet_offload_device() generator = torch.Generator(device=device).manual_seed(0)