浏览代码

fix gallery black image issue

w-e-w 1 年之前
父节点
当前提交
ee9d487081
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      modules/txt2img.py

+ 1 - 1
modules/txt2img.py

@@ -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