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(processor): restore missing update of session

Browse Source
This commit is contained in:
psychedelicious 2024-05-24 09:26:33 +10:00
parent 6c79be7dc3
commit 3c41c67d13
1 changed files with 5 additions and 1 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

@ -148,6 +148,9 @@ class DefaultSessionRunner(SessionRunnerBase):
graph_execution_state_id=queue_item.session.id, output_path=stats_path graph_execution_state_id=queue_item.session.id, output_path=stats_path
) )
# Update the queue item with the completed session
self._services.session_queue.set_queue_item_session(queue_item.item_id, queue_item.session)
# Send complete event # Send complete event
self._services.events.emit_graph_execution_complete( self._services.events.emit_graph_execution_complete(
queue_batch_id=queue_item.batch_id, queue_batch_id=queue_item.batch_id,
@ -399,9 +402,10 @@ class DefaultSessionProcessor(SessionProcessorBase):
# Non-fatal error in processor # Non-fatal error in processor
self._invoker.services.logger.error(f"Non-fatal error in session processor: {exc_type.__name__}") self._invoker.services.logger.error(f"Non-fatal error in session processor: {exc_type.__name__}")
self._invoker.services.logger.error(stacktrace) self._invoker.services.logger.error(stacktrace)
# Cancel the queue item
if queue_item is not None: if queue_item is not None:
# Update the queue item with the completed session
self._invoker.services.session_queue.set_queue_item_session(queue_item.item_id, queue_item.session) self._invoker.services.session_queue.set_queue_item_session(queue_item.item_id, queue_item.session)
# And cancel the queue item with an error
self._invoker.services.session_queue.cancel_queue_item(queue_item.item_id, error=stacktrace) self._invoker.services.session_queue.cancel_queue_item(queue_item.item_id, error=stacktrace)
for callback in self._on_non_fatal_processor_error_callbacks: for callback in self._on_non_fatal_processor_error_callbacks: