Allow use of refiner from img2img outputs from base

bghira · Jul 6, 2023 · e42d557 · e42d557
1 parent b23a000
commit e42d557
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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,