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

Interrupt after current generation

1 files changed, 7 insertions, 0 deletions

diff --git a/modules/shared_state.py b/modules/shared_state.py
index a68789cc..532fdcd8 100644
--- a/modules/shared_state.py
+++ b/modules/shared_state.py
@@ -12,6 +12,7 @@ log = logging.getLogger(__name__)
 class State:

     skipped = False

     interrupted = False

+    interrupted_next = False

     job = ""

     job_no = 0

     job_count = 0

@@ -79,6 +80,10 @@ class State:
         self.interrupted = True

         log.info("Received interrupt request")

 

+    def interrupt_next(self):

+        self.interrupted_next = True

+        log.info("Received interrupt request, interrupt after current job")

+

     def nextjob(self):

         if shared.opts.live_previews_enable and shared.opts.show_progress_every_n_steps == -1:

             self.do_set_current_image()

@@ -91,6 +96,7 @@ class State:
         obj = {

             "skipped": self.skipped,

             "interrupted": self.interrupted,

+            "interrupted_next": self.interrupted_next,

             "job": self.job,

             "job_count": self.job_count,

             "job_timestamp": self.job_timestamp,

@@ -114,6 +120,7 @@ class State:
         self.id_live_preview = 0

         self.skipped = False

         self.interrupted = False

+        self.interrupted_next = False

         self.textinfo = None

         self.job = job

         devices.torch_gc()