From e42d5579b1eb245579bdc4cf63e7b2bc1df65d77 Mon Sep 17 00:00:00 2001
From: bghira <bghira@users.github.com>
Date: Thu, 6 Jul 2023 14:19:00 -0700
Subject: [PATCH] Allow use of refiner from img2img outputs from base

---
 discord_tron_client/classes/image_manipulation/pipeline.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/discord_tron_client/classes/image_manipulation/pipeline.py b/discord_tron_client/classes/image_manipulation/pipeline.py
index e80e986c..4032113e 100644
--- a/discord_tron_client/classes/image_manipulation/pipeline.py
+++ b/discord_tron_client/classes/image_manipulation/pipeline.py
@@ -287,8 +287,8 @@ def _run_pipeline(
                         generator=torch_generators,
                     ).images
                 if use_latent_result:
-                    preprocessed_images = self._refiner_pipeline(
-                        images=preprocessed_images,
+                    new_image = self._refiner_pipeline(
+                        images=new_image,
                         user_config=user_config,
                         prompt=positive_prompt,
                         negative_prompt=negative_prompt,