80 lines
2.6 KiB
Python
80 lines
2.6 KiB
Python
SD_XL_BASE_RATIOS = {
|
|
"0.25":(512, 2048), # new
|
|
"0.26":(512, 1984), # new
|
|
"0.27":(512, 1920), # new
|
|
"0.28":(512, 1856), # new
|
|
"0.32":(576, 1792), # new
|
|
"0.33": (576, 1728), # new
|
|
"0.35": (576, 1664), # new
|
|
"0.4":(640, 1600), # new
|
|
"0.42": (640, 1536), # new
|
|
"0.48":(704, 1472), # new
|
|
"0.5": (704, 1408),
|
|
"0.52": (704, 1344),
|
|
"0.57": (768, 1344),
|
|
"0.6": (768, 1280),
|
|
"0.68": (832, 1216),
|
|
"0.72": (832, 1152),
|
|
"0.78": (896, 1152),
|
|
"0.82": (896, 1088),
|
|
"0.88": (960, 1088),
|
|
"0.94": (960, 1024),
|
|
"1.0": (1024, 1024),
|
|
"1.07": (1024, 960),
|
|
"1.13": (1088, 960),
|
|
"1.21": (1088, 896),
|
|
"1.29": (1152, 896),
|
|
"1.38": (1152, 832),
|
|
"1.46": (1216, 832),
|
|
"1.67": (1280, 768),
|
|
"1.75": (1344, 768),
|
|
"1.91": (1344, 704), # not in training value but in Stability-AI/generative-models/scripts/demo/sampling.py
|
|
"2.0": (1408, 704),
|
|
"2.09": (1472, 704),
|
|
"2.4": (1536, 640),
|
|
"2.5": (1600, 640),
|
|
"2.89": (1664, 576),
|
|
"3.0": (1728, 576),
|
|
"3.11":(1792, 576), # new
|
|
"3.62":(1856, 512), # new
|
|
"3.75":(1920, 512), # new
|
|
"3.88":(1984, 512), # new
|
|
"4.0":(2048, 512), # new
|
|
}
|
|
|
|
target_sizes_show = [f"{k}:{v}" for k, v in SD_XL_BASE_RATIOS.items()]
|
|
|
|
|
|
def get_SDXL_best_size(image_size = None, ratio = None):
|
|
"""
|
|
input a tuple such as get_SDXL_best_size((1200, 900)or input a float num such as get_SDXL_best_size(ratio=1.3)
|
|
return a tuple for SDXL such as (1152, 832)
|
|
"""
|
|
best_size = None
|
|
if image_size:
|
|
if image_size[0] > 0 and image_size[1] > 0:
|
|
ratio = image_size[0] / image_size[1] # w, h = image_size
|
|
if ratio:
|
|
target_sizes = [v for _, v in SD_XL_BASE_RATIOS.items()]
|
|
min_diff = float('inf') # a variable to store the minimum difference
|
|
for target_size in target_sizes:
|
|
target_ratio = target_size[0] / target_size[1]
|
|
diff = abs(ratio - target_ratio)
|
|
if diff < min_diff:
|
|
min_diff = diff
|
|
best_size = target_size
|
|
return best_size
|
|
|
|
# Test the function
|
|
def test1():
|
|
print(get_SDXL_best_size((800, 800))) # Output (1024, 1024)
|
|
print(get_SDXL_best_size((1200, 900))) # Output (1152, 896)
|
|
print(get_SDXL_best_size((600, 800))) # Output (896, 1152) # (832, 1152)
|
|
print(get_SDXL_best_size((700, 500))) # Output (1216, 832) # (1152, 832)
|
|
print(get_SDXL_best_size((1080, 1920))) # Output (768, 1344)
|
|
print(get_SDXL_best_size((1200, 500))) # Output (1536, 640)
|
|
print(get_SDXL_best_size(ratio=1.3)) # Output (1152, 896)
|
|
print(get_SDXL_best_size(ratio=1)) # Output (1024, 1024)
|
|
|
|
# test1()
|
|
# print(target_sizes_show) |