fix gallery black image issue

This commit is contained in:
w-e-w 2024-01-13 02:23:16 +09:00
parent cfb90a938e
commit ee9d487081
1 changed files with 1 additions and 1 deletions

View File

@ -93,7 +93,7 @@ def txt2img_upscale(id_task: str, request: gr.Request, gallery, gallery_index, g
new_gallery.extend(processed.images) new_gallery.extend(processed.images)
else: else:
fake_image = Image.new(mode="RGB", size=(1, 1)) fake_image = Image.new(mode="RGB", size=(1, 1))
fake_image.already_saved_as = image["name"] fake_image.already_saved_as = image["name"].rsplit('?', 1)[0]
new_gallery.append(fake_image) new_gallery.append(fake_image)
geninfo["infotexts"][gallery_index] = processed.info geninfo["infotexts"][gallery_index] = processed.info