remove debug print
This commit is contained in:
parent
1d4560cf36
commit
de13d2d6ef
|
@ -272,22 +272,20 @@ def run_inpaint_pipeline(model, provider, scheduler, prompt, negative_prompt, cf
|
||||||
rng = np.random.RandomState(seed)
|
rng = np.random.RandomState(seed)
|
||||||
|
|
||||||
extra = 256
|
extra = 256
|
||||||
(full_source, full_mask, full_noise, full_dims) = expand_image(source_image, mask_image, (extra, extra, extra, extra))
|
full_source, full_mask, full_noise, full_dims = expand_image(source_image, mask_image, (extra, extra, extra, extra))
|
||||||
|
full_width, full_height = full_dims
|
||||||
full_source.save('%s-source' % (output))
|
|
||||||
full_noise.save('%s-noise' % (output))
|
|
||||||
|
|
||||||
image = pipe(
|
image = pipe(
|
||||||
prompt,
|
prompt,
|
||||||
generator=rng,
|
generator=rng,
|
||||||
guidance_scale=cfg,
|
guidance_scale=cfg,
|
||||||
height=height,
|
height=full_height,
|
||||||
image=full_source,
|
image=full_source,
|
||||||
latents=latents,
|
latents=latents,
|
||||||
mask_image=full_mask,
|
mask_image=full_mask,
|
||||||
negative_prompt=negative_prompt,
|
negative_prompt=negative_prompt,
|
||||||
num_inference_steps=steps,
|
num_inference_steps=steps,
|
||||||
width=width,
|
width=full_width,
|
||||||
).images[0]
|
).images[0]
|
||||||
|
|
||||||
image.save(output)
|
image.save(output)
|
||||||
|
|
Loading…
Reference in New Issue