apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): explicitly delete pipeline results after saving

This commit is contained in:
Sean Sube 2023-01-21 22:28:13 -06:00
parent 9e26ee5b85
commit dddadfc9a2
1 changed files with 20 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
api/onnx_web/pipeline.py
View File

@ -106,7 +106,7 @@ def run_txt2img_pipeline(
latents = get_latents_from_seed(params.seed, size) latents = get_latents_from_seed(params.seed, size)
rng = np.random.RandomState(params.seed) rng = np.random.RandomState(params.seed)
image = pipe( result = pipe(
params.prompt, params.prompt,
height=size.height, height=size.height,
width=size.width, width=size.width,
@ -115,7 +115,8 @@ def run_txt2img_pipeline(
latents=latents, latents=latents,
negative_prompt=params.negative_prompt, negative_prompt=params.negative_prompt,
num_inference_steps=params.steps, num_inference_steps=params.steps,
).images[0] )
image = result.images[0]
if upscale.faces or upscale.scale > 1: if upscale.faces or upscale.scale > 1:
image = upscale_resrgan(ctx, upscale, image) image = upscale_resrgan(ctx, upscale, image)
@ -123,6 +124,9 @@ def run_txt2img_pipeline(
dest = safer_join(ctx.output_path, output) dest = safer_join(ctx.output_path, output)
image.save(dest) image.save(dest)
del image
del result
print('saved txt2img output: %s' % (dest)) print('saved txt2img output: %s' % (dest))
@ -139,7 +143,7 @@ def run_img2img_pipeline(
rng = np.random.RandomState(params.seed) rng = np.random.RandomState(params.seed)
image = pipe( result = pipe(
params.prompt, params.prompt,
generator=rng, generator=rng,
guidance_scale=params.cfg, guidance_scale=params.cfg,
@ -147,7 +151,8 @@ def run_img2img_pipeline(
negative_prompt=params.negative_prompt, negative_prompt=params.negative_prompt,
num_inference_steps=params.steps, num_inference_steps=params.steps,
strength=strength, strength=strength,
).images[0] )
image = result.images[0]
if upscale.faces or upscale.scale > 1: if upscale.faces or upscale.scale > 1:
image = upscale_resrgan(ctx, upscale, image) image = upscale_resrgan(ctx, upscale, image)
@ -155,6 +160,9 @@ def run_img2img_pipeline(
dest = safer_join(ctx.output_path, output) dest = safer_join(ctx.output_path, output)
image.save(dest) image.save(dest)
del image
del result
print('saved img2img output: %s' % (dest)) print('saved img2img output: %s' % (dest))
@ -192,7 +200,7 @@ def run_inpaint_pipeline(
mask_image.save(safer_join(ctx.output_path, 'last-mask.png')) mask_image.save(safer_join(ctx.output_path, 'last-mask.png'))
noise_image.save(safer_join(ctx.output_path, 'last-noise.png')) noise_image.save(safer_join(ctx.output_path, 'last-noise.png'))
image = pipe( result = pipe(
params.prompt, params.prompt,
generator=rng, generator=rng,
guidance_scale=params.cfg, guidance_scale=params.cfg,
@ -203,7 +211,8 @@ def run_inpaint_pipeline(
negative_prompt=params.negative_prompt, negative_prompt=params.negative_prompt,
num_inference_steps=params.steps, num_inference_steps=params.steps,
width=size.width, width=size.width,
).images[0] )
image = result.images[0]
if image.size == source_image.size: if image.size == source_image.size:
image = ImageChops.blend(source_image, image, strength) image = ImageChops.blend(source_image, image, strength)
@ -216,6 +225,9 @@ def run_inpaint_pipeline(
dest = safer_join(ctx.output_path, output) dest = safer_join(ctx.output_path, output)
image.save(dest) image.save(dest)
del image
del result
print('saved inpaint output: %s' % (dest)) print('saved inpaint output: %s' % (dest))
@ -232,4 +244,6 @@ def run_upscale_pipeline(
dest = safer_join(ctx.output_path, output) dest = safer_join(ctx.output_path, output)
image.save(dest) image.save(dest)
del image
print('saved img2img output: %s' % (dest)) print('saved img2img output: %s' % (dest))