diff --git a/ldm/invoke/generator/inpaint.py b/ldm/invoke/generator/inpaint.py
index 2ff0ca248a..bf63223016 100644
--- a/ldm/invoke/generator/inpaint.py
+++ b/ldm/invoke/generator/inpaint.py
@@ -189,6 +189,9 @@ class Inpaint(Img2Img):
 
         self.enable_image_debugging = enable_image_debugging
 
+        self.inpaint_width = inpaint_width
+        self.inpaint_height = inpaint_height
+
         if isinstance(init_image, PIL.Image.Image):
             self.pil_image = init_image.copy()
 
@@ -292,10 +295,6 @@ class Inpaint(Img2Img):
 
             result = self.sample_to_image(samples)
 
-            # Resize if necessary
-            if inpaint_width and inpaint_height:
-                result = result.resize(self.pil_image.size)
-
             # Seam paint if this is our first pass (seam_size set to 0 during seam painting)
             if seam_size > 0:
                 old_image = self.pil_image or init_image
@@ -335,6 +334,10 @@ class Inpaint(Img2Img):
         gen_result = super().sample_to_image(samples).convert('RGB')
         debug_image(gen_result, "gen_result", debug_status=self.enable_image_debugging)
 
+        # Resize if necessary
+        if self.inpaint_width and self.inpaint_height:
+            gen_result = gen_result.resize(self.pil_image.size)
+
         if self.pil_image is None or self.pil_mask is None:
             return gen_result