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

When an exception happens within the session processor loop, record and move on

Browse Source
This commit is contained in:
Brandon Rising 2023-09-21 11:10:25 -04:00
parent 9e7d829906
commit 883e9973ec
1 changed files with 23 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -92,7 +92,7 @@ class DefaultSessionProcessor(SessionProcessorBase):
self.__invoker.services.logger self.__invoker.services.logger
while not stop_event.is_set(): while not stop_event.is_set():
poll_now_event.clear() poll_now_event.clear()
try:
# do not dequeue if there is already a session running # do not dequeue if there is already a session running
if self.__queue_item is None and resume_event.is_set(): if self.__queue_item is None and resume_event.is_set():
queue_item = self.__invoker.services.session_queue.dequeue() queue_item = self.__invoker.services.session_queue.dequeue()
@ -116,6 +116,9 @@ class DefaultSessionProcessor(SessionProcessorBase):
continue continue
except Exception as e: except Exception as e:
self.__invoker.services.logger.error(f"Error in session processor: {e}") self.__invoker.services.logger.error(f"Error in session processor: {e}")
continue
except Exception as e:
self.__invoker.services.logger.error(f"Fatal Error in session processor: {e}")
pass pass
finally: finally:
stop_event.clear() stop_event.clear()