From 6c7068b5bc1a95243bf136f5b1ddb3e26ac8733a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jukka=20Sepp=C3=A4nen?=
 <40791699+kijai@users.noreply.github.com>
Date: Fri, 22 Nov 2024 22:38:51 +0200
Subject: [PATCH] Update nodes.py

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

diff --git a/nodes.py b/nodes.py
index 853b254..ac88912 100644
--- a/nodes.py
+++ b/nodes.py
@@ -361,7 +361,7 @@ class CogVideoImageEncodeFunInP:
 
         mask = torch.zeros_like(masked_image_latents[:, :, :1, :, :])
         if end_image is not None:
-            mask[:, -1, :, :, :] = vae_scaling_factor
+            mask[:, -1, :, :, :] = 0
         mask[:, 0, :, :, :] = vae_scaling_factor
 
         final_latents = masked_image_latents * vae_scaling_factor
@@ -971,4 +971,4 @@ NODE_DISPLAY_NAME_MAPPINGS = {
     "CogVideoLatentPreview": "CogVideo LatentPreview",
     "CogVideoXTorchCompileSettings": "CogVideo TorchCompileSettings",
     "CogVideoImageEncodeFunInP": "CogVideo ImageEncode FunInP",
-    }
\ No newline at end of file
+    }