Fixed an issue where img2img generation could not be interrupted in the latest webui

pull/48/head
s9roll7 2023-04-06 12:02:56 +09:00
parent 1e91627c22
commit de9d914e74
1 changed files with 5 additions and 0 deletions

View File

@ -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