apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): repair progress for API pipelines

This commit is contained in:
Sean Sube 2023-06-30 22:00:11 -05:00
parent 2d10252564
commit d9f251c88b
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 16 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
api/onnx_web/chain/upscale_highres.py
View File

@ -56,15 +56,11 @@ def upscale_highres(
chain=chain,
)
chain.append(
(
chain.stage(
blend_img2img,
StageParams(),
{
"overlap": params.overlap,
"strength": highres.strength,
},
)
overlap=params.overlap,
strength=highres.strength,
)
return chain(

16
api/onnx_web/diffusers/run.py
View File

@ -79,7 +79,8 @@ def run_txt2img_pipeline(
)
# run and save
image = chain(job, server, params, None)
progress = job.get_progress_callback()
image = chain(job, server, params, None, callback=progress)
_prompt_pairs, loras, inversions = parse_prompt(params)
dest = save_image(
@ -167,8 +168,9 @@ def run_img2img_pipeline(
)
# run and append the filtered source
progress = job.get_progress_callback()
images = [
chain(job, server, params, source),
chain(job, server, params, source, callback=progress),
]
if source_filter is not None and source_filter != "none":
@ -247,7 +249,8 @@ def run_inpaint_pipeline(
)
# run and save
image = chain(job, server, params, source)
progress = job.get_progress_callback()
image = chain(job, server, params, source, callback=progress)
_prompt_pairs, loras, inversions = parse_prompt(params)
dest = save_image(
@ -312,7 +315,9 @@ def run_upscale_pipeline(
)
# run and save
image = chain(job, server, params, source)
progress = job.get_progress_callback()
image = chain(job, server, params, source, callback=progress)
_prompt_pairs, loras, inversions = parse_prompt(params)
dest = save_image(
server,
@ -359,7 +364,8 @@ def run_blend_pipeline(
)
# run and save
image = chain(job, server, params, sources[0])
progress = job.get_progress_callback()
image = chain(job, server, params, sources[0], callback=progress)
dest = save_image(server, outputs[0], image, params, size, upscale=upscale)
# clean up