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

make pause/resume work in multithreaded environment

Browse Source
This commit is contained in:
Lincoln Stein 2024-04-16 16:55:56 -04:00
parent 89f8326c0b
commit eaadc55c7d
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -181,7 +181,9 @@ class DefaultSessionProcessor(SessionProcessorBase):
while True: while True:
# Outer try block. Any error here is a fatal processor error # Outer try block. Any error here is a fatal processor error
try: try:
self._resume_event.wait()
session = self._session_worker_queue.get() session = self._session_worker_queue.get()
if self._cancel_event.is_set(): if self._cancel_event.is_set():
if session.item_id in self._sessions_to_cancel: if session.item_id in self._sessions_to_cancel:
continue continue