From 6b4f147a078a4c866741cd69fb28659d553c715a Mon Sep 17 00:00:00 2001 From: missionfloyd Date: Thu, 21 Dec 2023 19:03:25 -0700 Subject: [PATCH] Fix img2img interrogate --- modules/ui.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/modules/ui.py b/modules/ui.py index 5be344d220a..0634cdb1e09 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -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)