extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-04-16 23:48:25 +02:00
Code Issues Packages Projects Releases Wiki Activity

Missing a 'r' in progress

Browse Source
This commit is contained in:
Marc-Andre Ferland 2022-10-22 01:23:39 -04:00
parent 8a10fcf7ea
commit cd6d49860f
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ui/sd_internal/runtime.py
View File

@ -403,7 +403,7 @@ def update_temp_img(req, x_samples):
return partial_images return partial_images
# Build and return the apropriate generator for do_mk_img # Build and return the apropriate generator for do_mk_img
def get_image_progess_generator(req, extra_props=None): def get_image_progress_generator(req, extra_props=None):
if not req.stream_progress_updates: if not req.stream_progress_updates:
def empty_callback(x_samples, i): return x_samples def empty_callback(x_samples, i): return x_samples
return empty_callback return empty_callback
@ -563,7 +563,7 @@ def do_mk_img(req: Request):
thread_data.modelFS.to(thread_data.device) thread_data.modelFS.to(thread_data.device)
n_steps = req.num_inference_steps if req.init_image is None else t_enc n_steps = req.num_inference_steps if req.init_image is None else t_enc
img_callback = get_image_progess_generator(req, {"total_steps": n_steps}) img_callback = get_image_progress_generator(req, {"total_steps": n_steps})
# run the handler # run the handler
try: try: