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

Better error handling with cache.put

Browse Source
This commit is contained in:
Marc-Andre Ferland 2022-10-15 04:08:17 -04:00
parent 1b32423881
commit d3b28c42e6
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
ui/sd_internal/task_manager.py
View File

@ -113,7 +113,9 @@ class TaskCache():
self._base[key] = (
self._get_ttl_time(ttl), value
)
except Exception:
except Exception as e:
print(str(e))
print(traceback.format_exc())
return False
else:
return True
@ -283,6 +285,7 @@ def render(req : ImageRequest):
r.stream_image_progress = False
new_task = RenderTask(r)
task_cache.put(r.session_id, new_task, TASK_TTL)
tasks_queue.put(new_task)
return new_task
if task_cache.put(r.session_id, new_task, TASK_TTL):
tasks_queue.put(new_task)
return new_task
raise RuntimeError('Failed to add task to cache.')