mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-12-06 10:31:17 -08:00
Code Issues Projects Releases Packages Wiki Activity

store the last generated result

Browse source
This commit is contained in:
siutin 2023-02-06 03:53:05 +08:00
parent dbca512154
commit 9407f1731a
2 changed files with 11 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

10
modules/progress.py
View file

@ -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")