From da78367bdf4f50adcc90ecf1fe1102e548cba625 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Wed, 26 Mar 2025 17:36:59 +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 3fc2f99..23ffce7 100644
--- a/nodes.py
+++ b/nodes.py
@@ -418,7 +418,7 @@ class WanVideoModelLoader:
                 model_variant = "i2v_720"
         elif model_type == "t2v":
             model_variant = "14B"
-        if dim == 5120:
+        if dim == 1536:
             model_variant = "1_3B"
         log.info(f"Model variant detected: {model_variant}")