From 4bac36356aa61ea6fcda2c893eedf694b9cb8c29 Mon Sep 17 00:00:00 2001
From: blessedcoolant <54517381+blessedcoolant@users.noreply.github.com>
Date: Thu, 31 Aug 2023 06:33:09 +1200
Subject: [PATCH] fix: Create SDXL Refiner Create Mask only in inpaint &
 outpaint

---
 .../nodes/util/graphBuilders/addVAEToGraph.ts | 25 +++++++++++--------
 1 file changed, 15 insertions(+), 10 deletions(-)

diff --git a/invokeai/frontend/web/src/features/nodes/util/graphBuilders/addVAEToGraph.ts b/invokeai/frontend/web/src/features/nodes/util/graphBuilders/addVAEToGraph.ts
index 37b79ad621..b274f8fbba 100644
--- a/invokeai/frontend/web/src/features/nodes/util/graphBuilders/addVAEToGraph.ts
+++ b/invokeai/frontend/web/src/features/nodes/util/graphBuilders/addVAEToGraph.ts
@@ -149,16 +149,21 @@ export const addVAEToGraph = (
   }
 
   if (shouldUseSDXLRefiner) {
-    graph.edges.push({
-      source: {
-        node_id: isAutoVae ? modelLoaderNodeId : VAE_LOADER,
-        field: isAutoVae && isOnnxModel ? 'vae_decoder' : 'vae',
-      },
-      destination: {
-        node_id: SDXL_REFINER_INPAINT_CREATE_MASK,
-        field: 'vae',
-      },
-    });
+    if (
+      graph.id === SDXL_CANVAS_INPAINT_GRAPH ||
+      graph.id === SDXL_CANVAS_OUTPAINT_GRAPH
+    ) {
+      graph.edges.push({
+        source: {
+          node_id: isAutoVae ? modelLoaderNodeId : VAE_LOADER,
+          field: isAutoVae && isOnnxModel ? 'vae_decoder' : 'vae',
+        },
+        destination: {
+          node_id: SDXL_REFINER_INPAINT_CREATE_MASK,
+          field: 'vae',
+        },
+      });
+    }
   }
 
   if (vae && metadataAccumulator) {