fix(api): correct output names for inpaint pipelines
This commit is contained in:
parent
8f7ef6dfce
commit
c2d45b03dc
|
@ -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,33 +416,23 @@ 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):
|
||||||
|
|
||||||
for image, thumbnail in zip(images.as_images(), thumbnail_names):
|
|
||||||
thumbnails.append(
|
|
||||||
save_image(
|
save_image(
|
||||||
server,
|
server,
|
||||||
thumbnail,
|
thumbnail,
|
||||||
image,
|
image,
|
||||||
)
|
)
|
||||||
)
|
|
||||||
|
|
||||||
images.outputs = outputs
|
|
||||||
images.thumbnails = thumbnails
|
|
||||||
|
|
||||||
# clean up
|
# clean up
|
||||||
run_gc([worker.get_device()])
|
run_gc([worker.get_device()])
|
||||||
|
|
Loading…
Reference in New Issue