apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

move test to inpaint endpoint

This commit is contained in:
Sean Sube 2023-01-14 13:24:25 -06:00
parent a3029c30a6
commit 9ed3a3fd3e
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
api/onnx_web/serve.py
View File

@ -267,8 +267,6 @@ def run_img2img_pipeline(model, provider, scheduler, prompt, negative_prompt, cf
rng = np.random.RandomState(seed)
input_image = expand_image(input_image, (256, 256, 256, 256))
image = pipe(
prompt,
generator=rng,
@ -290,14 +288,18 @@ def run_inpaint_pipeline(model, provider, scheduler, prompt, negative_prompt, cf
latents = get_latents_from_seed(seed, width, height)
rng = np.random.RandomState(seed)
full_mask = expand_image(mask_image, (256, 256, 256, 256))
full_source = expand_image(source_image, (256, 256, 256, 256))
# full_source.thumbnail((512, 512))
image = pipe(
prompt,
generator=rng,
guidance_scale=cfg,
height=height,
image=source_image,
image=full_source,
latents=latents,
mask_image=mask_image,
mask_image=full_mask,
negative_prompt=negative_prompt,
num_inference_steps=steps,
width=width,