From 3ce9640497139b14462910b0dbf2be1df855c1d6 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Fri, 9 May 2025 20:13:08 +0300
Subject: [PATCH] Update nodes.py

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

diff --git a/nodes.py b/nodes.py
index df200a9..2b3a9ec 100644
--- a/nodes.py
+++ b/nodes.py
@@ -330,7 +330,7 @@ class HyVideoModelLoader:
         in_channels = sd["img_in.proj.weight"].shape[1]
         if in_channels == 16 and "i2v" in model.lower():
             i2v_condition_type = "token_replace"
-        elif in_channels == 16 and not "i2v" in model.lower():
+        elif in_channels == 16 and "custom" in model.lower():
             i2v_condition_type = "reference"
         else:
             i2v_condition_type = "latent_concat"