diff options
| author | siutin <osiutino@gmail.com> | 2023-02-06 07:57:26 +0000 |
|---|---|---|
| committer | siutin <osiutino@gmail.com> | 2023-03-30 09:20:09 +0000 |
| commit | e0b58527ff040f9c547ea45b5fcf1bfb7ab23cdd (patch) | |
| tree | 1ea75a39d9ebeab51601d5d30f28abacbaa11eb1 /modules/progress.py | |
| parent | 4242e194e417ec5008d09ec6d756594ac65f77bd (diff) | |
| download | stable-diffusion-webui-gfx803-e0b58527ff040f9c547ea45b5fcf1bfb7ab23cdd.tar.gz stable-diffusion-webui-gfx803-e0b58527ff040f9c547ea45b5fcf1bfb7ab23cdd.tar.bz2 stable-diffusion-webui-gfx803-e0b58527ff040f9c547ea45b5fcf1bfb7ab23cdd.zip | |
use condition to wait for result
Diffstat (limited to 'modules/progress.py')
| -rw-r--r-- | modules/progress.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/modules/progress.py b/modules/progress.py index 36963c92..1947c0fd 100644 --- a/modules/progress.py +++ b/modules/progress.py @@ -6,6 +6,7 @@ import gradio as gr from pydantic import BaseModel, Field
from typing import List
+from modules import call_queue
from modules.shared import opts
import modules.shared as shared
@@ -57,8 +58,9 @@ def restore_progress_call(task_tag): else:
t_task = current_task
- while t_task != last_task_id:
- time.sleep(2.5)
+ with call_queue.queue_lock_condition:
+ call_queue.queue_lock_condition.wait_for(lambda: t_task == last_task_id)
+
return last_task_result
|