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