From 87bb4d8f6e66d3bb21db60a6197b77a30e3cacb2 Mon Sep 17 00:00:00 2001
From: blessedcoolant <54517381+blessedcoolant@users.noreply.github.com>
Date: Tue, 29 Aug 2023 04:52:41 +1200
Subject: [PATCH] fix: Seamless not working with SDXL on Canvas

---
 .../graphBuilders/addSeamlessToLinearGraph.ts | 96 ++++++++++---------
 1 file changed, 53 insertions(+), 43 deletions(-)

diff --git a/invokeai/frontend/web/src/features/nodes/util/graphBuilders/addSeamlessToLinearGraph.ts b/invokeai/frontend/web/src/features/nodes/util/graphBuilders/addSeamlessToLinearGraph.ts
index e506d2e5bb..e4816d9c87 100644
--- a/invokeai/frontend/web/src/features/nodes/util/graphBuilders/addSeamlessToLinearGraph.ts
+++ b/invokeai/frontend/web/src/features/nodes/util/graphBuilders/addSeamlessToLinearGraph.ts
@@ -2,17 +2,18 @@ import { RootState } from 'app/store/store';
 import { SeamlessModeInvocation } from 'services/api/types';
 import { NonNullableGraph } from '../../types/types';
 import {
-  CANVAS_IMAGE_TO_IMAGE_GRAPH,
-  CANVAS_TEXT_TO_IMAGE_GRAPH,
+  CANVAS_COHERENCE_DENOISE_LATENTS,
+  CANVAS_INPAINT_GRAPH,
+  CANVAS_OUTPAINT_GRAPH,
   DENOISE_LATENTS,
-  IMAGE_TO_IMAGE_GRAPH,
   SDXL_CANVAS_IMAGE_TO_IMAGE_GRAPH,
+  SDXL_CANVAS_INPAINT_GRAPH,
+  SDXL_CANVAS_OUTPAINT_GRAPH,
   SDXL_CANVAS_TEXT_TO_IMAGE_GRAPH,
   SDXL_DENOISE_LATENTS,
   SDXL_IMAGE_TO_IMAGE_GRAPH,
   SDXL_TEXT_TO_IMAGE_GRAPH,
   SEAMLESS,
-  TEXT_TO_IMAGE_GRAPH,
 } from './constants';
 
 export const addSeamlessToLinearGraph = (
@@ -34,7 +35,9 @@ export const addSeamlessToLinearGraph = (
 
   if (
     graph.id === SDXL_TEXT_TO_IMAGE_GRAPH ||
-    graph.id === SDXL_IMAGE_TO_IMAGE_GRAPH
+    graph.id === SDXL_IMAGE_TO_IMAGE_GRAPH ||
+    graph.id === SDXL_CANVAS_TEXT_TO_IMAGE_GRAPH ||
+    graph.id === SDXL_CANVAS_IMAGE_TO_IMAGE_GRAPH
   ) {
     denoisingNodeId = SDXL_DENOISE_LATENTS;
   }
@@ -51,47 +54,54 @@ export const addSeamlessToLinearGraph = (
       )
   );
 
+  graph.edges.push(
+    {
+      source: {
+        node_id: modelLoaderNodeId,
+        field: 'unet',
+      },
+      destination: {
+        node_id: SEAMLESS,
+        field: 'unet',
+      },
+    },
+    {
+      source: {
+        node_id: modelLoaderNodeId,
+        field: 'vae',
+      },
+      destination: {
+        node_id: SEAMLESS,
+        field: 'vae',
+      },
+    },
+    {
+      source: {
+        node_id: SEAMLESS,
+        field: 'unet',
+      },
+      destination: {
+        node_id: denoisingNodeId,
+        field: 'unet',
+      },
+    }
+  );
+
   if (
-    graph.id === TEXT_TO_IMAGE_GRAPH ||
-    graph.id === IMAGE_TO_IMAGE_GRAPH ||
-    graph.id === SDXL_TEXT_TO_IMAGE_GRAPH ||
-    graph.id === SDXL_IMAGE_TO_IMAGE_GRAPH ||
-    graph.id === CANVAS_TEXT_TO_IMAGE_GRAPH ||
-    graph.id === CANVAS_IMAGE_TO_IMAGE_GRAPH ||
-    graph.id === SDXL_CANVAS_TEXT_TO_IMAGE_GRAPH ||
-    graph.id == SDXL_CANVAS_IMAGE_TO_IMAGE_GRAPH
+    graph.id == CANVAS_INPAINT_GRAPH ||
+    graph.id === CANVAS_OUTPAINT_GRAPH ||
+    graph.id === SDXL_CANVAS_INPAINT_GRAPH ||
+    graph.id === SDXL_CANVAS_OUTPAINT_GRAPH
   ) {
-    graph.edges.push(
-      {
-        source: {
-          node_id: modelLoaderNodeId,
-          field: 'unet',
-        },
-        destination: {
-          node_id: SEAMLESS,
-          field: 'unet',
-        },
+    graph.edges.push({
+      source: {
+        node_id: SEAMLESS,
+        field: 'unet',
       },
-      {
-        source: {
-          node_id: modelLoaderNodeId,
-          field: 'vae',
-        },
-        destination: {
-          node_id: SEAMLESS,
-          field: 'vae',
-        },
+      destination: {
+        node_id: CANVAS_COHERENCE_DENOISE_LATENTS,
+        field: 'unet',
       },
-      {
-        source: {
-          node_id: SEAMLESS,
-          field: 'unet',
-        },
-        destination: {
-          node_id: denoisingNodeId,
-          field: 'unet',
-        },
-      }
-    );
+    });
   }
 };