Modify apply_overlay for inpainting

Author: root

src/diffusers/image_processor.py

@@ -659,13 +659,11 @@ def apply_overlay(
         overlay the inpaint output to the original image
         """
 
-        width, height = image.width, image.height
-
-        init_image = self.resize(init_image, width=width, height=height)
-        mask = self.resize(mask, width=width, height=height)
+        width, height = init_image.width, init_image.height
 
         init_image_masked = PIL.Image.new("RGBa", (width, height))
         init_image_masked.paste(init_image.convert("RGBA").convert("RGBa"), mask=ImageOps.invert(mask.convert("L")))
+        
         init_image_masked = init_image_masked.convert("RGBA")
 
         if crop_coords is not None:
@@ -676,14 +674,15 @@ def apply_overlay(
             image = self.resize(image, height=h, width=w, resize_mode="crop")
             base_image.paste(image, (x, y))
             image = base_image.convert("RGB")
-
-        image = image.convert("RGBA")
+            
+        image = image.convert("RGBA") 
         image.alpha_composite(init_image_masked)
         image = image.convert("RGB")
 
         return image
 
 
+
 class VaeImageProcessorLDM3D(VaeImageProcessor):
     """
     Image processor for VAE LDM3D.