Spaces:
Running
on
Zero
Running
on
Zero
File size: 2,278 Bytes
3d5837a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 |
from typing import Any, Mapping
SDXL_SUPPORTED_RESOLUTIONS = [
(1024, 1024, 1.0),
(1152, 896, 1.2857142857142858),
(896, 1152, 0.7777777777777778),
(1216, 832, 1.4615384615384615),
(832, 1216, 0.6842105263157895),
(1344, 768, 1.75),
(768, 1344, 0.5714285714285714),
(1536, 640, 2.4),
(640, 1536, 0.4166666666666667),
]
class SDXLResolution:
@classmethod
def INPUT_TYPES(cls) -> Mapping[str, Any]:
return {
"required": {
"resolution": (
[f"{res[0]}x{res[1]}" for res in SDXL_SUPPORTED_RESOLUTIONS],
)
}
}
RETURN_TYPES = ("INT", "INT")
RETURN_NAMES = ("width", "height")
FUNCTION = "op"
CATEGORY = "math/graphics"
def op(self, resolution: str) -> tuple[int, int]:
width, height = resolution.split("x")
return (int(width), int(height))
class NearestSDXLResolution:
@classmethod
def INPUT_TYPES(cls) -> Mapping[str, Any]:
return {"required": {"image": ("IMAGE",)}}
RETURN_TYPES = ("INT", "INT")
RETURN_NAMES = ("width", "height")
FUNCTION = "op"
CATEGORY = "math/graphics"
def op(self, image) -> tuple[int, int]:
image_width = image.size()[2]
image_height = image.size()[1]
print(f"Input image resolution: {image_width}x{image_height}")
image_ratio = image_width / image_height
differences = [
(abs(image_ratio - resolution[2]), resolution)
for resolution in SDXL_SUPPORTED_RESOLUTIONS
]
smallest = None
for difference in differences:
if smallest is None:
smallest = difference
else:
if difference[0] < smallest[0]:
smallest = difference
if smallest is not None:
width = smallest[1][0]
height = smallest[1][1]
else:
width = 1024
height = 1024
print(f"Selected SDXL resolution: {width}x{height}")
return (width, height)
NODE_CLASS_MAPPINGS = {
"CM_SDXLResolution": SDXLResolution,
"CM_NearestSDXLResolution": NearestSDXLResolution,
}
|