From 8aa13d5dce2789a7d0bd802e6d62453b3c380496 Mon Sep 17 00:00:00 2001
From: Anthony Fu <anthonyfu117@hotmail.com>
Date: Mon, 16 Oct 2023 14:12:18 +0800
Subject: Interrupt after current generation

---
 modules/processing.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'modules/processing.py')

diff --git a/modules/processing.py b/modules/processing.py
index 40598f5c..e7eecd66 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -819,7 +819,7 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
             if state.skipped:
                 state.skipped = False
 
-            if state.interrupted:
+            if state.interrupted or state.interrupted_next:
                 break
 
             sd_models.reload_model_weights()  # model can be changed for example by refiner
-- 
cgit v1.2.3