diff --git a/invokeai/app/services/session_processor/session_processor_default.py b/invokeai/app/services/session_processor/session_processor_default.py index e7caabdf903..61270e08795 100644 --- a/invokeai/app/services/session_processor/session_processor_default.py +++ b/invokeai/app/services/session_processor/session_processor_default.py @@ -86,6 +86,12 @@ async def _on_queue_event(self, event: FastAPIEvent) -> None: self._poll_now() elif event_name == "batch_enqueued": 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: if not self._resume_event.is_set():