lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-01-26 19:09:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Hiresfix quickbutton insert option (#2325)

Browse Source 
add gallery insert option for hiresfix quickbutton, and select the new image in the new gallery
This commit is contained in:
DenOfEquity
2024-11-15 12:16:29 +00:00
committed by GitHub
parent 19a9a78c9b
commit 95021aebec
2 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
modules/shared_options.py
View File

@@ -319,6 +319,7 @@ options_templates.update(options_section(('ui_gallery', "Gallery", "ui"), {
"sd_webui_modal_lightbox_toolbar_opacity": OptionInfo(0.9, "Full page image viewer: tool bar opacity", gr.Slider, {"minimum": 0.0, "maximum": 1, "step": 0.01}, onchange=shared.reload_gradio_theme).info('for mouse only').needs_reload_ui(),
"gallery_height": OptionInfo("", "Gallery height", gr.Textbox).info("can be any valid CSS value, for example 768px or 20em").needs_reload_ui(),
"open_dir_button_choice": OptionInfo("Subdirectory", "What directory the [📂] button opens", gr.Radio, {"choices": ["Output Root", "Subdirectory", "Subdirectory (even temp dir)"]}),
"hires_button_gallery_insert": OptionInfo(False, "Insert [✨] hires button results into gallery").info("Default: original image will be replaced"),
}))
options_templates.update(options_section(('ui_alternatives', "UI alternatives", "ui"), {

16
modules/txt2img.py
View File

@@ -102,16 +102,22 @@ def txt2img_upscale_function(id_task: str, request: gr.Request, gallery, gallery
shared.total_tqdm.clear()
insert = getattr(shared.opts, 'hires_button_gallery_insert', False)
new_gallery = []
for i, image in enumerate(gallery):
if insert or i != gallery_index:
new_gallery.append(image)
if i == gallery_index:
new_gallery.extend(processed.images)
else:
new_gallery.append(image)
new_index = gallery_index
if insert:
new_index += 1
geninfo["infotexts"].insert(new_index, processed.info)
else:
geninfo["infotexts"][gallery_index] = processed.info
geninfo["infotexts"][gallery_index] = processed.info
return new_gallery, json.dumps(geninfo), plaintext_to_html(processed.info), plaintext_to_html(processed.comments, classname="comments")
return gr.Gallery(new_gallery, selected_index=new_index), json.dumps(geninfo), plaintext_to_html(processed.info), plaintext_to_html(processed.comments, classname="comments")
def txt2img_function(id_task: str, request: gr.Request, *args):