apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

pass progress to txt2img and img2img

This commit is contained in:
Sean Sube 2023-02-12 12:38:06 -06:00
parent 2f6a3afddb
commit feb7e7d57b
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/diffusion/run.py
View File

@ -65,7 +65,7 @@ def run_txt2img_pipeline(
image = result.images[0] image = result.images[0]
image = run_upscale_correction( image = run_upscale_correction(
job, server, StageParams(), params, image, upscale=upscale job, server, StageParams(), params, image, upscale=upscale, callback=progress,
) )
dest = save_image(server, output, image) dest = save_image(server, output, image)
@ -123,7 +123,7 @@ def run_img2img_pipeline(
image = result.images[0] image = result.images[0]
image = run_upscale_correction( image = run_upscale_correction(
job, server, StageParams(), params, image, upscale=upscale job, server, StageParams(), params, image, upscale=upscale, callback=progress,
) )
dest = save_image(server, output, image) dest = save_image(server, output, image)