Keep the task alive during step callbacks. Thanks Madrang

2023-04-21 20:59:14 +05:30 · 2023-04-21 20:59:14 +05:30 · f7235cf82c
commit f7235cf82c
parent eb16296873
1 changed files with 3 additions and 0 deletions
--- a/ui/easydiffusion/task_manager.py
+++ b/ui/easydiffusion/task_manager.py
@ -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)