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

Merge branch 'beta' of https://github.com/cmdr2/stable-diffusion-ui.git into multi-gpu

Browse Source
This commit is contained in:
Marc-Andre Ferland
2022-10-18 13:21:26 -04:00
parent 5e461e9b6b 946dfdf7b8
commit cfd6751777
37 changed files with 737 additions and 711 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
ui/sd_internal/runtime.py
View File

@ -478,14 +478,18 @@ def do_mk_img(req: Request):
thread_data.modelFS.to(thread_data.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 = []