diff --git a/ldm/invoke/generator/omnibus.py b/ldm/invoke/generator/omnibus.py index 7eaefee0c63..d6b5b164ab8 100644 --- a/ldm/invoke/generator/omnibus.py +++ b/ldm/invoke/generator/omnibus.py @@ -167,6 +167,8 @@ def sample_to_image(self, samples)->Image.Image: if self.pil_image is None or self.pil_mask is None: return gen_result + if self.pil_image.size != self.pil_mask.size: + return gen_result corrected_result = super(Img2Img, self).repaste_and_color_correct(gen_result, self.pil_image, self.pil_mask, self.mask_blur_radius)