mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2026-03-07 14:21:40 -08:00
Code Issues Projects Releases Packages Wiki Activity

Fix img2img interrogate

Browse source
This commit is contained in:
missionfloyd 2023-12-21 19:03:25 -07:00
parent 92b33344bf
commit 6b4f147a07
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
modules/ui.py
View file

@ -109,10 +109,11 @@ def resize_from_to_html(width, height, scale_by):
def process_interrogate(interrogation_function, mode, ii_input_dir, ii_output_dir, *ii_singles):
if mode in {0, 1, 3, 4}:
return [interrogation_function(ii_singles[mode]), None]
mode = int(mode)
if mode in (0, 1, 3, 4):
return [interrogation_function(ii_singles[mode]["composite"]), None]
elif mode == 2:
return [interrogation_function(ii_singles[mode]["image"]), None]
return [interrogation_function(ii_singles[mode]["composite"]), None]
elif mode == 5:
assert not shared.cmd_opts.hide_ui_dir_config, "Launched with --hide-ui-dir-config, batch img2img disabled"
images = shared.listfiles(ii_input_dir)