From feb7e7d57bbcd1b7dd5e90d4875165f0cd50b8bb Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 12 Feb 2023 12:38:06 -0600 Subject: [PATCH] pass progress to txt2img and img2img --- api/onnx_web/diffusion/run.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/onnx_web/diffusion/run.py b/api/onnx_web/diffusion/run.py index 6c69eabe..225bd3e5 100644 --- a/api/onnx_web/diffusion/run.py +++ b/api/onnx_web/diffusion/run.py @@ -65,7 +65,7 @@ def run_txt2img_pipeline( image = result.images[0] 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) @@ -123,7 +123,7 @@ def run_img2img_pipeline( image = result.images[0] 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)