From 9681c83c1d1bcb27140d032b878ee3a65eee7239 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Tue, 12 Nov 2024 00:58:00 +0200
Subject: [PATCH] Update nodes.py

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

diff --git a/nodes.py b/nodes.py
index 5a5345a..1f1a80b 100644
--- a/nodes.py
+++ b/nodes.py
@@ -842,7 +842,7 @@ class CogVideoSampler:
             context_options is not None
         ), "1.0 I2V model can only do 49 frames"
         if image_cond_latents is not None:
-            assert image_cond_latents.shape[0] == 1, "Image condition latents must be a single latent"
+            assert image_cond_latents["samples"].shape[0] == 1, "Image condition latents must be a single latent"
             assert "I2V" in pipeline.get("model_name", ""), "Image condition latents only supported for I2V models"
         else:
             assert "I2V" not in pipeline.get("model_name", ""), "Image condition latents required for I2V models"