fix extras tab showing original images instead of upscales

1 files changed, 4 insertions, 1 deletions

diff --git a/modules/extras.py b/modules/extras.py
index ffae7d67..38d6ec48 100644
--- a/modules/extras.py
+++ b/modules/extras.py
@@ -36,6 +36,7 @@ def run_extras(image, image_folder, gfpgan_visibility, codeformer_visibility, co
 

     outpath = opts.outdir_samples or opts.outdir_extras_samples

 

+    outputs = []

     for image in imageArr:

         existing_pnginfo = image.info or {}

 

@@ -91,7 +92,9 @@ def run_extras(image, image_folder, gfpgan_visibility, codeformer_visibility, co
 

         images.save_image(image, path=outpath, basename="", seed=None, prompt=None, extension=opts.samples_format, info=info, short_filename=True, no_prompt=True, grid=False, pnginfo_section_name="extras", existing_info=existing_pnginfo)

 

-    return imageArr, plaintext_to_html(info), ''

+        outputs.append(image)

+

+    return outputs, plaintext_to_html(info), ''

 

 

 def run_pnginfo(image):