Mirrors
/
stable-diffusion-webui
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

allowe hr pass to return multiple images

This commit is contained in:
w-e-w 2024-01-13 02:22:32 +09:00
parent 92501d4f80
commit cfb90a938e
1 changed files with 4 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
modules/txt2img.py
View File

@ -88,18 +88,13 @@ def txt2img_upscale(id_task: str, request: gr.Request, gallery, gallery_index, g
new_gallery = []
for i, image in enumerate(gallery):
fake_image = Image.new(mode="RGB", size=(1, 1))
if i == gallery_index:
already_saved_as = getattr(processed.images[0], 'already_saved_as', None)
if already_saved_as is not None:
fake_image.already_saved_as = already_saved_as
else:
fake_image = processed.images[0]
geninfo["infotexts"][gallery_index: gallery_index+1] = processed.infotexts
new_gallery.extend(processed.images)
else:
fake_image = Image.new(mode="RGB", size=(1, 1))
fake_image.already_saved_as = image["name"]
new_gallery.append(fake_image)
new_gallery.append(fake_image)
geninfo["infotexts"][gallery_index] = processed.info