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
This commit is contained in:
Manuel Schmid 2024-08-11 17:05:24 +02:00 committed by GitHub
parent b0d16a3aa7
commit 14895ebb13
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions

View File

@ -1223,6 +1223,8 @@ def worker():
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