From e0b58527ff040f9c547ea45b5fcf1bfb7ab23cdd Mon Sep 17 00:00:00 2001
From: siutin <osiutino@gmail.com>
Date: Mon, 6 Feb 2023 15:57:26 +0800
Subject: [PATCH] use condition to wait for result

---
 modules/call_queue.py | 2 +-
 modules/progress.py   | 6 ++++--
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/modules/call_queue.py b/modules/call_queue.py
index 30ac26bc6..9888109ec 100644
--- a/modules/call_queue.py
+++ b/modules/call_queue.py
@@ -7,7 +7,7 @@ import time
 from modules import shared, progress
 
 queue_lock = threading.Lock()
-
+queue_lock_condition = threading.Condition(lock=queue_lock)
 
 def wrap_queued_call(func):
     def f(*args, **kwargs):
diff --git a/modules/progress.py b/modules/progress.py
index 36963c92e..1947c0fdd 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