From de9d914e74a21e178ab860060a9caad88f9415a2 Mon Sep 17 00:00:00 2001 From: s9roll7 Date: Thu, 6 Apr 2023 12:02:56 +0900 Subject: [PATCH] Fixed an issue where img2img generation could not be interrupted in the latest webui --- scripts/custom_script.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/scripts/custom_script.py b/scripts/custom_script.py index 170ef7f..661d7e5 100644 --- a/scripts/custom_script.py +++ b/scripts/custom_script.py @@ -940,6 +940,11 @@ class Script(scripts.Script): # img2img for img, mask, controlnet_input_img, face_coords, prompts in zip(imgs, masks, controlnet_input_imgs, face_coords_dict.values(), prompts_dict.values()): + # Generation cancelled. + if shared.state.interrupted: + print("Generation cancelled.") + break + image = Image.open(img) mask_image = Image.open(mask) if mask else None