pass progress to txt2img and img2img
This commit is contained in:
parent
2f6a3afddb
commit
feb7e7d57b
|
@ -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)
|
||||||
|
|
Loading…
Reference in New Issue