Update nodes.py

2024-11-22 22:38:51 +02:00
parent 895d3b83a4
commit 6c7068b5bc
1 changed files with 2 additions and 2 deletions
@@ -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",
-    }
+    }