Update nodes.py

2024-08-30 22:13:26 +03:00
parent c1efd95a03
commit bbbb6e9514
1 changed files with 1 additions and 1 deletions
@@ -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)