diff options
| author | siutin <osiutino@gmail.com> | 2023-02-05 19:53:05 +0000 |
|---|---|---|
| committer | siutin <osiutino@gmail.com> | 2023-03-30 09:20:09 +0000 |
| commit | 9407f1731aa8c112ffc0efaa611a76f7fead3d0c (patch) | |
| tree | 28498d648e226e8e6401a7fb16c81204d02626e5 | |
| parent | dbca512154341bb13e1b15d207176f2d403aff30 (diff) | |
| download | stable-diffusion-webui-gfx803-9407f1731aa8c112ffc0efaa611a76f7fead3d0c.tar.gz stable-diffusion-webui-gfx803-9407f1731aa8c112ffc0efaa611a76f7fead3d0c.tar.bz2 stable-diffusion-webui-gfx803-9407f1731aa8c112ffc0efaa611a76f7fead3d0c.zip | |
store the last generated result
| -rw-r--r-- | modules/call_queue.py | 1 | ||||
| -rw-r--r-- | modules/progress.py | 10 |
2 files changed, 11 insertions, 0 deletions
diff --git a/modules/call_queue.py b/modules/call_queue.py index 92097c15..30ac26bc 100644 --- a/modules/call_queue.py +++ b/modules/call_queue.py @@ -37,6 +37,7 @@ def wrap_gradio_gpu_call(func, extra_outputs=None): res = func(*args, **kwargs)
finally:
progress.finish_task(id_task)
+ progress.set_last_task_result(id_task, res)
shared.state.end()
diff --git a/modules/progress.py b/modules/progress.py index 05032ac5..27a336ad 100644 --- a/modules/progress.py +++ b/modules/progress.py @@ -37,6 +37,16 @@ def finish_task(id_task): def add_task_to_queue(id_job):
pending_tasks[id_job] = time.time()
+last_task_id = None
+last_task_result = None
+
+def set_last_task_result(id_job, result):
+ global last_task_id
+ global last_task_result
+
+ last_task_id = id_job
+ last_task_result = result
+
class CurrentTaskResponse(BaseModel):
current_task: str = Field(default=None, title="Task ID", description="id of the current progress task")
|