extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2024-12-26 00:49:17 +01:00
Code Issues 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():
global current_state_error
task_cache.keep(id(task), TASK_TTL)
session_cache.keep(task.task_data.session_id, TASK_TTL)
if (
isinstance(current_state_error, SystemExit)
or isinstance(current_state_error, StopAsyncIteration)