From afe6639b9cd11ec1baae0d9bebf0878f84a19411 Mon Sep 17 00:00:00 2001
From: blessedcoolant <54517381+blessedcoolant@users.noreply.github.com>
Date: Fri, 23 Feb 2024 04:54:16 +0530
Subject: [PATCH] fix: outpaint result not getting pasted back correctly

---
 invokeai/app/invocations/image.py                              | 2 --
 .../src/features/nodes/util/graph/buildCanvasOutpaintGraph.ts  | 3 ++-
 .../features/nodes/util/graph/buildCanvasSDXLOutpaintGraph.ts  | 3 ++-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/invokeai/app/invocations/image.py b/invokeai/app/invocations/image.py
index ab4d49b26c..f8c3aad62b 100644
--- a/invokeai/app/invocations/image.py
+++ b/invokeai/app/invocations/image.py
@@ -963,9 +963,7 @@ class IAICanvasPasteBackInvocation(BaseInvocation, WithMetadata, WithBoard):
         target_image = context.images.get_pil(self.target_image.image_name)
         mask = self._prepare_mask(context.images.get_pil(self.mask.image_name))
 
-        # Merge the bands back together
         source_image.paste(target_image, (0, 0), mask)
 
         image_dto = context.images.save(image=source_image)
-
         return ImageOutput.build(image_dto)
diff --git a/invokeai/frontend/web/src/features/nodes/util/graph/buildCanvasOutpaintGraph.ts b/invokeai/frontend/web/src/features/nodes/util/graph/buildCanvasOutpaintGraph.ts
index 4140c4c485..eb98b40802 100644
--- a/invokeai/frontend/web/src/features/nodes/util/graph/buildCanvasOutpaintGraph.ts
+++ b/invokeai/frontend/web/src/features/nodes/util/graph/buildCanvasOutpaintGraph.ts
@@ -176,6 +176,7 @@ export const buildCanvasOutpaintGraph = (
         is_intermediate: getIsIntermediate(state),
         board: getBoardField(state),
         use_cache: false,
+        mask_blur: canvasCoherenceEdgeSize,
       },
     },
     edges: [
@@ -468,7 +469,7 @@ export const buildCanvasOutpaintGraph = (
       },
       {
         source: {
-          node_id: LATENTS_TO_IMAGE,
+          node_id: INPAINT_IMAGE_RESIZE_DOWN,
           field: 'image',
         },
         destination: {
diff --git a/invokeai/frontend/web/src/features/nodes/util/graph/buildCanvasSDXLOutpaintGraph.ts b/invokeai/frontend/web/src/features/nodes/util/graph/buildCanvasSDXLOutpaintGraph.ts
index 6a664acb51..d181d1d3a2 100644
--- a/invokeai/frontend/web/src/features/nodes/util/graph/buildCanvasSDXLOutpaintGraph.ts
+++ b/invokeai/frontend/web/src/features/nodes/util/graph/buildCanvasSDXLOutpaintGraph.ts
@@ -176,6 +176,7 @@ export const buildCanvasSDXLOutpaintGraph = (
         is_intermediate: getIsIntermediate(state),
         board: getBoardField(state),
         use_cache: false,
+        mask_blur: canvasCoherenceEdgeSize,
       },
     },
     edges: [
@@ -477,7 +478,7 @@ export const buildCanvasSDXLOutpaintGraph = (
       },
       {
         source: {
-          node_id: LATENTS_TO_IMAGE,
+          node_id: INPAINT_IMAGE_RESIZE_DOWN,
           field: 'image',
         },
         destination: {