apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

run outpainting at original size

This commit is contained in:
Sean Sube 2023-01-14 16:31:47 -06:00
parent 23ddbc6b10
commit 899d41a38d
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/onnx_web/serve.py
View File

@ -268,24 +268,24 @@ def run_inpaint_pipeline(model, provider, scheduler, prompt, negative_prompt, cf
pipe = load_pipeline(OnnxStableDiffusionInpaintPipeline,
model, provider, scheduler)
latents = get_latents_from_seed(seed, width, height)
rng = np.random.RandomState(seed)
extra = 256
full_source, full_mask, full_noise, full_dims = expand_image(source_image, mask_image, (extra, extra, extra, extra))
full_width, full_height = full_dims
latents = get_latents_from_seed(seed, full_width, full_height)
image = pipe(
prompt,
generator=rng,
guidance_scale=cfg,
height=full_height,
height=height,
image=full_source,
latents=latents,
mask_image=full_mask,
negative_prompt=negative_prompt,
num_inference_steps=steps,
width=full_width,
width=width,
).images[0]
image.save(output)