Mirror/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2024-06-07 21:20:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: refactor

Browse Source
This commit is contained in:
Anthony Fu 2023-10-16 15:00:17 +08:00
parent 8aa13d5dce
commit 3d15e58b0a
2 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
modules/shared_state.py
View File

@ -77,12 +77,12 @@ class State:
log.info("Received skip request") log.info("Received skip request")
def interrupt(self): def interrupt(self):
if shared.opts.interrupt_after_current and self.job_count > 1: self.interrupted = True
self.interrupted_next = True log.info("Received interrupt request")
log.info("Received interrupt request, interrupt after current job")
else: def interrupt_next(self):
self.interrupted = True self.interrupted_next = True
log.info("Received interrupt request") log.info("Received interrupt request, interrupt after current job")
def nextjob(self): def nextjob(self):
if shared.opts.live_previews_enable and shared.opts.show_progress_every_n_steps == -1: if shared.opts.live_previews_enable and shared.opts.show_progress_every_n_steps == -1:

8
modules/ui.py
View File

@ -216,8 +216,14 @@ class Toprow:
outputs=[], outputs=[],
) )
def interrupt_fn():
if shared.state.job_count > 1 and shared.opts.interrupt_after_current:
shared.state.interrupt_next()
else:
shared.state.interrupt()
self.interrupt.click( self.interrupt.click(
fn=lambda: shared.state.interrupt(), fn=interrupt_fn,
inputs=[], inputs=[],
outputs=[], outputs=[],
) )