Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore(nodes): add comments for cancel state

Browse Source
This commit is contained in:
psychedelicious 2024-02-18 11:42:40 +11:00
parent 6a74048af8
commit 556f6aa174
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
invokeai/app/services/session_processor/session_processor_default.py
View File

@ -192,9 +192,20 @@ class DefaultSessionProcessor(SessionProcessorBase):
)
except KeyboardInterrupt:
# TODO(psyche): should we set the cancel event here and/or cancel the queue item?
pass
except CanceledException:
# When the user cancels the graph, we first set the cancel event. The event is checked
# between invocations, in this loop. Some invocations are long-running, and we need to
# be able to cancel them mid-execution.
#
# For example, denoising is a long-running invocation with many steps. A step callback
# is executed after each step. This step callback checks if the canceled event is set,
# then raises a CanceledException to stop execution immediately.
#
# When we get a CanceledException, we don't need to do anything - just pass and let the
# loop go to its next iteration, and the cancel event will be handled correctly.
pass
except Exception as e: