fix(api): explicitly delete pipeline results after saving
This commit is contained in:
parent
9e26ee5b85
commit
dddadfc9a2
|
@ -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))
|
||||||
|
|
Loading…
Reference in New Issue