apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): correct output names for inpaint pipelines

This commit is contained in:
Sean Sube 2024-01-29 20:56:25 -06:00
parent 8f7ef6dfce
commit c2d45b03dc
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 8 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -404,11 +404,9 @@ def run_inpaint_pipeline(
)
# custom version of save for full-res inpainting
output_names = make_output_names(server, worker.job, len(images))
outputs = []
images.outputs = make_output_names(server, worker.job, len(images))
for image, metadata, output in zip(
images.as_images(), images.metadata, output_names
images.as_images(), images.metadata, images.outputs
):
if full_res_inpaint:
if is_debug():
@ -418,34 +416,24 @@ def run_inpaint_pipeline(
image = original_source
image.paste(mini_image, box=adj_mask_border)
outputs.append(
save_image(
server,
output,
image,
metadata,
)
)
thumbnails = None
if params.thumbnail:
thumbnail_names = make_output_names(
images.thumbnails = make_output_names(
server, worker.job, len(images), suffix="thumbnail"
)
thumbnails = []
for image, thumbnail in zip(images.as_images(), thumbnail_names):
thumbnails.append(
save_image(
server,
thumbnail,
image,
)
for image, thumbnail in zip(images.as_images(), images.thumbnails):
save_image(
server,
thumbnail,
image,
)
images.outputs = outputs
images.thumbnails = thumbnails
# clean up
run_gc([worker.get_device()])