diff --git a/mikey_nodes.py b/mikey_nodes.py index 1b169de..6eda9a5 100644 --- a/mikey_nodes.py +++ b/mikey_nodes.py @@ -79,7 +79,12 @@ class EmptyLatentRatioCustom: def generate(self, width, height, batch_size=1): # solver - w, h = sdxl_size(width, height) + if width == 1 and height == 1: + w, h = 1024, 1024 + if f'{width}:{height}' in EmptyLatentRatioSelector.ratio_dict: + w, h = EmptyLatentRatioSelector.ratio_dict[f'{width}:{height}'] + else: + w, h = sdxl_size(width, height) latent = torch.zeros([batch_size, 4, h // 8, w // 8]) return ({"samples":latent}, ) @@ -183,4 +188,7 @@ NODE_CLASS_MAPPINGS = { 'Save Image With Prompt Data': SaveImagesMikey, 'Resize Image for SDXL': ResizeImageSDXL, 'VAE Decode 6GB SDXL (deprecated)': VAEDecode6GB, -} \ No newline at end of file +} +## TODO +# Resize Image and return the new width and height +# SDXL Ultimate Upscaler? \ No newline at end of file