Merge pull request #7 from GavChap/GavChap-patch-1

Update sizes.json

Author: GavChap

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-sd3latentselectres"
-description = "You'll get a new node called SD3 Latent Select Resolution, you can pick the x and y sizes from a list."
-version = "1.0.1"
+description = "You'll get a new node called SD3/Flux Latent Select Resolution, you can pick the x and y sizes from a list."
+version = "2.0.0"
 license = { file = "LICENSE" }
 
 [project.urls]
@@ -10,5 +10,5 @@ Repository = "https://github.com/GavChap/ComfyUI-SD3LatentSelectRes"
 
 [tool.comfy]
 PublisherId = "anymode"
-DisplayName = "ComfyUI-SD3LatentSelectRes"
+DisplayName = "SD3 and Flux Latent Select Resolution"
 Icon = ""

sd3latentselectres.py

@@ -14,6 +14,7 @@ def INPUT_TYPES(cls):
         return {
             'required': {
                 'size_selected': (cls.size_sizes,),
+                'flip_ratio': ("BOOLEAN", {"default": False}),
                 'batch_size': ("INT", {"default": 1, "min": 1, "max": 4096})
             }
         }
@@ -24,11 +25,16 @@ def INPUT_TYPES(cls):
     OUTPUT_NODE = True
     CATEGORY = "generate/sd3"
 
-    def return_res(self, size_selected, batch_size):
+    def return_res(self, size_selected, flip_ratio, batch_size):
         # Extract resolution name and dimensions using the key
         selected_info = self.size_dict[size_selected]
-        width = int(selected_info["width"])
-        height = int(selected_info["height"])
+        if (flip_ratio):
+            width = int(selected_info["height"])
+            height = int(selected_info["width"])    
+        else:
+            width = int(selected_info["width"])
+            height = int(selected_info["height"])    
+
         latent = torch.ones([batch_size, 16, height // 8, width // 8], device=self.device) * 0.0609
         return {"width": width}, {"height": height}, {"samples": latent},
 
@@ -48,7 +54,7 @@ def read_sizes():
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
-    "SD3LatentSelectRes": "SD3 Select Latent Resolution"
+    "SD3LatentSelectRes": "SD3/Flux Select Latent Resolution"
 }
 
 

sizes.json

@@ -1,19 +1,17 @@
 {
     "sizes": {
-        "1:1 [1024x1024 square]": {"width": 1024, "height":  1024},
-        "8:5 [1216x768 landscape]": {"width": 1216, "height":  768},
-        "4:3 [1152x896 landscape]": {"width": 1152, "height":  896},
-        "3:2 [1216x832 landscape]": {"width": 1216, "height":  832},
-        "7:5 [1176x840 landscape]": {"width": 1176, "height":  840},
-        "16:9 [1344x768 landscape]": {"width": 1344, "height":  768},
-        "21:9 [1536x640 landscape]": {"width": 1536, "height":  640},
-        "19:9 [1472x704 landscape]": {"width": 1472, "height":  704},
-        "3:4 [896x1152 portrait]": {"width": 896, "height":  1152},
-        "2:3 [832x1216 portrait]": {"width": 832, "height":  1216},
-        "5:7 [840x1176 portrait]": {"width": 840, "height":  1176},
-        "9:16 [768x1344 portrait]": {"width": 768, "height":  1344},
-        "9:21 [640x1536 portrait]": {"width": 640, "height":  1536},
-        "5:8 [768x1216 portrait]": {"width": 768, "height":  1216},
-        "9:19 [704x1472 portrait]": {"width": 704, "height":  1472}
+        "1MP - 1:1 [1024x1024]": {"width": 1024, "height": 1024},
+        "1MP - 8:5 [1216x768]": {"width": 1216, "height": 768},
+        "1MP - 4:3 [1152x896]": {"width": 1152, "height": 896},
+        "1MP - 3:2 [1216x832]": {"width": 1216, "height": 832},
+        "1MP - 7:5 [1176x840]": {"width": 1176, "height": 840},
+        "1MP - 16:9 [1344x768]": {"width": 1344, "height": 768},
+        "1MP - 21:9 [1536x640]": {"width": 1536, "height": 640},
+        "1MP - 19:9 [1472x704]": {"width": 1472, "height": 704},
+        "2MP - 1:1 [1408x1408]": {"width": 1408, "height": 1408},
+        "2MP - 3:2 [1728x1152]": {"width": 1728, "height": 1152},
+        "2MP - 4:3 [1664x1216]": {"width": 1664, "height": 1216},
+        "2MP - 16:9 [1920x1088]": {"width": 1920, "height": 1088},
+        "2MP - 21:9 [2176x960]": {"width": 2176, "height": 960}
     }
-}
+}