Merge pull request #13653 from antfu/feat/interrupted-end

Interrupt after current generation

1 files changed, 3 insertions, 3 deletions

diff --git a/scripts/loopback.py b/scripts/loopback.py
index 2d5feaf9..ad921269 100644
--- a/scripts/loopback.py
+++ b/scripts/loopback.py
@@ -95,7 +95,7 @@ class Script(scripts.Script):
                 processed = processing.process_images(p)

 

                 # Generation cancelled.

-                if state.interrupted:

+                if state.interrupted or state.interrupted_next:

                     break

 

                 if initial_seed is None:

@@ -122,8 +122,8 @@ class Script(scripts.Script):
 

             p.inpainting_fill = original_inpainting_fill

 

-            if state.interrupted:

-                    break

+            if state.interrupted or state.interrupted_next:

+                break

 

         if len(history) > 1:

             grid = images.image_grid(history, rows=1)