faisalhr1997 commited on
Commit
8074f11
1 Parent(s): c07abba

Upload SDXLAspectRatio.py

Browse files
Files changed (1) hide show
  1. SDXLAspectRatio.py +70 -0
SDXLAspectRatio.py ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # quick node to set SDXL-friendly aspect ratios in 1024^2
2
+ # by throttlekitty
3
+
4
+ class SDXLAspectRatio:
5
+ def __init__(self):
6
+ pass
7
+
8
+ @classmethod
9
+ def INPUT_TYPES(s):
10
+ return {
11
+ "required": {
12
+ "width": ("INT", {"default": 64, "min": 64, "max": 2048,}),
13
+ "height": ("INT", {"default": 64, "min": 64, "max": 2048}),
14
+ "aspectRatio": ([
15
+ "1:1 - 1024x1024 square",
16
+ "2:3 - 832x1216 portrait",
17
+ "3:4 - 896x1152 portrait",
18
+ "5:8 - 768x1216 portrait",
19
+ "9:16 - 768x1344 portrait",
20
+ "9:19 - 704x1472 portrait",
21
+ "9:21 - 640x1536 portrait",
22
+ "3:2 - 1216x832 landscape",
23
+ "4:3 - 1152x896 landscape",
24
+ "8:5 - 1216x768 landscape",
25
+ "16:9 - 1344x768 landscape",
26
+ "19:9 - 1472x704 landscape",
27
+ "21:9 - 1536x640 landscape"],)
28
+ }
29
+ }
30
+ RETURN_TYPES = ("INT", "INT")
31
+ RETURN_NAMES = ("Width", "Height")
32
+ FUNCTION = "SDXL_AspectRatio"
33
+ CATEGORY = "image"
34
+
35
+ def SDXL_AspectRatio(self, width, height, aspectRatio):
36
+ if aspectRatio == "1:1 - 1024x1024 square":
37
+ width, height = 1024, 1024
38
+ elif aspectRatio == "2:3 - 832x1216 portrait":
39
+ width, height = 832, 1216
40
+ elif aspectRatio == "3:4 - 896x1152 portrait":
41
+ width, height = 896, 1152
42
+ elif aspectRatio == "5:8 - 768x1216 portrait":
43
+ width, height = 768, 1216
44
+ elif aspectRatio == "9:16 - 768x1344 portrait":
45
+ width, height = 768, 1344
46
+ elif aspectRatio == "9:19 - 704x1472 portrait":
47
+ width, height = 704, 1472
48
+ elif aspectRatio == "9:21 - 640x1536 portrait":
49
+ width, height = 640, 1536
50
+ elif aspectRatio == "3:2 - 1216x832 landscape":
51
+ width, height = 1216, 832
52
+ elif aspectRatio == "4:3 - 1152x896 landscape":
53
+ width, height = 1152, 896
54
+ elif aspectRatio == "8:5 - 1216x768 landscape":
55
+ width, height = 1216, 768
56
+ elif aspectRatio == "16:9 - 1344x768 landscape":
57
+ width, height = 1344, 768
58
+ elif aspectRatio == "19:9 - 1472x704 landscape":
59
+ width, height = 1472, 704
60
+ elif aspectRatio == "21:9 - 1536x640 landscape":
61
+ width, height = 1536, 640
62
+ return(width, height)
63
+
64
+
65
+ NODE_CLASS_MAPPINGS = {
66
+ "SDXLAspectRatio": SDXLAspectRatio
67
+ }
68
+ NODE_DISPLAY_NAME_MAPPINGS = {
69
+ "SDXLAspectRatio": "SDXL Aspect Ratio"
70
+ }