Update nodes.py

2025-03-06 13:14:33 +02:00
parent 9fc32187e6
commit 4863a2af8d
1 changed files with 3 additions and 3 deletions
@@ -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")