From c2d45b03dc129c3d6c3cc4a01e557cb3448c1e23 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Mon, 29 Jan 2024 20:56:25 -0600 Subject: [PATCH] fix(api): correct output names for inpaint pipelines --- api/onnx_web/diffusers/run.py | 28 ++++++++-------------------- 1 file changed, 8 insertions(+), 20 deletions(-) diff --git a/api/onnx_web/diffusers/run.py b/api/onnx_web/diffusers/run.py index ea5cde79..65e9e339 100644 --- a/api/onnx_web/diffusers/run.py +++ b/api/onnx_web/diffusers/run.py @@ -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()])