apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

remove debug print

This commit is contained in:
Sean Sube 2023-01-14 16:27:34 -06:00
parent 1d4560cf36
commit de13d2d6ef
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
api/onnx_web/serve.py
View File

@ -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)