From 35571eb14dd518b113757f53c416c74017547a8e Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Tue, 13 Dec 2022 11:44:20 +0530 Subject: [PATCH] Don't hang the task if something other than the renderer fails (e.g. model loading) --- ui/sd_internal/task_manager.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/ui/sd_internal/task_manager.py b/ui/sd_internal/task_manager.py index 434394cf..a8c5a4b2 100644 --- a/ui/sd_internal/task_manager.py +++ b/ui/sd_internal/task_manager.py @@ -372,6 +372,8 @@ def thread_render(device): session_cache.keep(task.request.session_id, TASK_TTL) except Exception as e: task.error = e + task.response = {"status": 'failed', "detail": str(task.error)} + task.buffer_queue.put(json.dumps(task.response)) print(traceback.format_exc()) continue finally: