StableDiffusion/WebUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

use condition to wait for result

Browse Source
This commit is contained in:
siutin 2023-02-06 15:57:26 +08:00
parent 4242e194e4
commit e0b58527ff
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/call_queue.py
View File

@ -7,7 +7,7 @@ import time
from modules import shared, progress from modules import shared, progress
queue_lock = threading.Lock() queue_lock = threading.Lock()
queue_lock_condition = threading.Condition(lock=queue_lock)
def wrap_queued_call(func): def wrap_queued_call(func):
def f(*args, **kwargs): def f(*args, **kwargs):

6
modules/progress.py
View File

@ -6,6 +6,7 @@ import gradio as gr
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
from typing import List from typing import List
from modules import call_queue
from modules.shared import opts from modules.shared import opts
import modules.shared as shared import modules.shared as shared
@ -57,8 +58,9 @@ def restore_progress_call(task_tag):
else: else:
t_task = current_task t_task = current_task
while t_task != last_task_id: with call_queue.queue_lock_condition:
time.sleep(2.5) call_queue.queue_lock_condition.wait_for(lambda: t_task == last_task_id)
return last_task_result return last_task_result