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

Only cancel session processor if current generating queue item is cancelled

Browse Source
This commit is contained in:
brandonrising
2024-03-28 14:22:15 -04:00
committed by psychedelicious
parent e8eb9fd533
commit 98cc9b963c
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -70,8 +70,13 @@ class DefaultSessionProcessor(SessionProcessorBase):
async def _on_queue_event(self, event: FastAPIEvent) -> None: async def _on_queue_event(self, event: FastAPIEvent) -> None:
event_name = event[1]["event"] event_name = event[1]["event"]
if event_name == "session_canceled" or event_name == "queue_cleared":
# These both mean we should cancel the current session. if event_name == "session_canceled" and self._queue_item\
and self._queue_item.item_id == event[1]["data"]["queue_item_id"]:
self._cancel_event.set()
self._poll_now()
elif event_name == "queue_cleared" and self._queue_item\
and self._queue_item.queue_id == event[1]["data"]["queue_id"]:
self._cancel_event.set() self._cancel_event.set()
self._poll_now() self._poll_now()
elif event_name == "batch_enqueued": elif event_name == "batch_enqueued":