holm/easydiffusion
1
0
Fork 0
forked from extern/easydiffusion
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Keep the task alive during step callbacks. Thanks Madrang

Browse Source
This commit is contained in:
cmdr2 2023-04-21 20:59:14 +05:30
parent eb16296873
commit f7235cf82c
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ui/easydiffusion/task_manager.py
View File

@ -317,6 +317,9 @@ def thread_render(device):
def step_callback(): def step_callback():
global current_state_error global current_state_error
task_cache.keep(id(task), TASK_TTL)
session_cache.keep(task.task_data.session_id, TASK_TTL)
if ( if (
isinstance(current_state_error, SystemExit) isinstance(current_state_error, SystemExit)
or isinstance(current_state_error, StopAsyncIteration) or isinstance(current_state_error, StopAsyncIteration)