Interrupt after current generation

This commit is contained in:
Anthony Fu
2023-10-16 14:12:18 +08:00
parent 861cbd5636
commit 8aa13d5dce
7 changed files with 17 additions and 8 deletions

View File

@@ -49,7 +49,7 @@ def process_batch(p, input_dir, output_dir, inpaint_mask_dir, args, to_scale=Fal
if state.skipped:
state.skipped = False
if state.interrupted:
if state.interrupted or state.interrupted_next:
break
try: