From 6c46304eb833c2b78c27cddae98aff0d14fd4574 Mon Sep 17 00:00:00 2001
From: blessedcoolant <54517381+blessedcoolant@users.noreply.github.com>
Date: Tue, 25 Jul 2023 23:20:54 +1200
Subject: [PATCH] fix: Replug Image To Latents VAE back in the Refiner graph
 for img2img

---
 .../util/graphBuilders/buildSDXLRefinerGraph.ts   | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/invokeai/frontend/web/src/features/nodes/util/graphBuilders/buildSDXLRefinerGraph.ts b/invokeai/frontend/web/src/features/nodes/util/graphBuilders/buildSDXLRefinerGraph.ts
index 113b6f0be7..d7535deeed 100644
--- a/invokeai/frontend/web/src/features/nodes/util/graphBuilders/buildSDXLRefinerGraph.ts
+++ b/invokeai/frontend/web/src/features/nodes/util/graphBuilders/buildSDXLRefinerGraph.ts
@@ -2,6 +2,7 @@ import { RootState } from 'app/store/store';
 import { MetadataAccumulatorInvocation } from 'services/api/types';
 import { NonNullableGraph } from '../../types/types';
 import {
+  IMAGE_TO_LATENTS,
   LATENTS_TO_IMAGE,
   METADATA_ACCUMULATOR,
   SDXL_MODEL_LOADER,
@@ -73,12 +74,18 @@ export const addSDXLRefinerToGraph = (
     scheduler: refinerScheduler,
     denoising_start: refinerStart,
   };
-  // graph.nodes[LATENTS_TO_IMAGE] = {
-  //   type: 'l2i',
-  //   id: LATENTS_TO_IMAGE,
-  // };
 
   graph.edges.push(
+    {
+      source: {
+        node_id: SDXL_MODEL_LOADER,
+        field: 'vae',
+      },
+      destination: {
+        node_id: IMAGE_TO_LATENTS,
+        field: 'vae',
+      },
+    },
     {
       source: {
         node_id: SDXL_REFINER_MODEL_LOADER,