From 4863a2af8d9b00fc1ddb1829c3cf445a78485349 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Thu, 6 Mar 2025 13:14:33 +0200
Subject: [PATCH] Update nodes.py

---
 nodes.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/nodes.py b/nodes.py
index eea2438..d39637e 100644
--- a/nodes.py
+++ b/nodes.py
@@ -1530,11 +1530,11 @@ class HyVideoGetClosestBucketSize:
     CATEGORY = "HunyuanVideoWrapper"
 
     def encode(self, image, base_size):
-        if base_size == "720p":
+        if base_size == "720":
             bucket_hw_base_size = 960
-        elif base_size == "540p":
+        elif base_size == "540":
             bucket_hw_base_size = 720
-        elif base_size == "360p":
+        elif base_size == "360":
             bucket_hw_base_size = 480
         else:
             raise NotImplementedError(f"Base size {base_size} not implemented")