fix fix

utils.py

@@ -146,7 +146,7 @@ def generate_novelai_image(
         payload['action'] = "infill"
         payload['model'] = model.replace('-preview', '') + '-inpainting'
         mask = inp_img['layers'][0]
-        mask = mask.resize((mask.size[0]//8, mask.size[1]//8), resample=PIL.Image.NEAREST).resize(mask.size, resample=PIL.Image.NEAREST).point(lambda x: 0 if x<255 else 255)
+        mask = mask.resize((width, height)).resize((mask.size[0]//8, mask.size[1]//8), resample=PIL.Image.NEAREST).resize(mask.size, resample=PIL.Image.NEAREST).point(lambda x: 0 if x<255 else 255)
         payload['parameters']['mask'] = image2base64(mask)
         payload['parameters']['image'] = image2base64(inp_img['background'])
         payload['parameters']['extra_noise_seed'] = seed
@@ -157,6 +157,13 @@ def generate_novelai_image(
         payload['parameters']['extra_noise_seed'] = seed
         payload["parameters"]['noise'] = i2i_noise
     if chr_image != None:
+        scale = chr_image.size[0]/chr_image.size[1]
+        if scale < 0.8165:
+            chr_image = chr_image.resize((1024, 1536))
+        elif scale < 1.2247:
+            chr_image = chr_image.resize((1427, 1427))
+        else:
+            chr_image = chr_image.resize((1536, 1024))
         payload['parameters']['director_reference_images'] = [image2base64(chr_image)]
         payload['parameters']['director_reference_descriptions'] = [{'caption': {'base_caption': 'character&style' if style_aware else 'character', 'char_captions': []}, 'legacy_uc': False}]
         payload['parameters']['director_reference_secondary_strength_values'] = [1 - fidelity]