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