From 9baf1003663928079c56e7ed4f84832d96463795 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Fri, 22 Nov 2024 23:13:14 +0200
Subject: [PATCH] fix 2b image2vid

---
 nodes.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/nodes.py b/nodes.py
index ac88912..3819b54 100644
--- a/nodes.py
+++ b/nodes.py
@@ -606,7 +606,12 @@ class CogVideoSampler:
         mm.soft_empty_cache()
 
         model_name = model.get("model_name", "")
-        supports_image_conds = True if "I2V" in model_name or "interpolation" in model_name.lower() or "fun" in model_name.lower() else False
+        supports_image_conds = True if (
+        "I2V" in model_name or 
+        "interpolation" in model_name.lower() or 
+        "fun" in model_name.lower() or
+        "img2vid" in model_name.lower()
+        ) else False
         if "fun" in model_name.lower() and not ("pose" in model_name.lower() or "control" in model_name.lower()) and image_cond_latents is not None:
             assert image_cond_latents["mask"] is not None, "For fun inpaint models use CogVideoImageEncodeFunInP"
             fun_mask = image_cond_latents["mask"]