extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-08-10 08:07:47 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix the broken 'time remaining' counter

Browse Source
This commit is contained in:
cmdr2
2022-10-18 16:12:17 +05:30
parent b23bc4a5b6
commit 83d6c3ba88
2 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
ui/sd_internal/runtime.py
View File

@ -405,14 +405,18 @@ def do_mk_img(req: Request):
modelFS.to(device)
partial_x_samples = None
last_callback_time = -1
def img_callback(x_samples, i):
nonlocal partial_x_samples
nonlocal partial_x_samples, last_callback_time
partial_x_samples = x_samples
if req.stream_progress_updates:
n_steps = req.num_inference_steps if req.init_image is None else t_enc
progress = {"step": i, "total_steps": n_steps}
step_time = time.time() - last_callback_time if last_callback_time != -1 else -1
last_callback_time = time.time()
progress = {"step": i, "total_steps": n_steps, "step_time": step_time}
if req.stream_image_progress and i % 5 == 0:
partial_images = []