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

fix(queue): poll queue on finished queue item

Browse Source 
When a queue item is finished (completed, canceled, failed), immediately poll the queue for the next queue item.

Closes #6189
This commit is contained in:
psychedelicious 2024-04-10 13:36:29 +10:00
parent 651c0b39b1
commit b18442ded4
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -86,6 +86,12 @@ class DefaultSessionProcessor(SessionProcessorBase):
self._poll_now() self._poll_now()
elif event_name == "batch_enqueued": elif event_name == "batch_enqueued":
self._poll_now() self._poll_now()
elif event_name == "queue_item_status_changed" and event[1]["data"]["queue_item"]["status"] in [
"completed",
"failed",
"canceled",
]:
self._poll_now()
def resume(self) -> SessionProcessorStatus: def resume(self) -> SessionProcessorStatus:
if not self._resume_event.is_set(): if not self._resume_event.is_set():