hotfix: yield enhance_input_image to correctly preview debug masks (#…

…3497) sort images starts from index <images_to_enhance_count>, which is 1 if enhance_input_image has been provided
lllyasviel · Aug 11, 2024 · 14895eb · 14895eb · mhd978 · Aug 25, 2024
1 parent b0d16a3
commit 14895eb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/modules/async_worker.py b/modules/async_worker.py
@@ -1223,6 +1223,8 @@ def handler(async_task: AsyncTask):
             height, width, _ = async_task.enhance_input_image.shape
             # input image already provided, processing is skipped
             steps = 0
+            yield_result(async_task, async_task.enhance_input_image, current_progress, async_task.black_out_nsfw, False,
+                         async_task.disable_intermediate_results)
 
         all_steps = steps * async_task.image_number