apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): add latents to inpaint, remove strength

This commit is contained in:
Sean Sube 2023-01-08 18:16:51 -06:00
parent 182ce6de90
commit 131cff6ba4
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
api/onnx_web/serve.py
View File

@ -345,21 +345,23 @@ def inpaint():
mask_image = Image.open(BytesIO(mask_file.read())).convert('RGB') mask_image = Image.open(BytesIO(mask_file.read())).convert('RGB')
mask_image.thumbnail((default_width, default_height)) mask_image.thumbnail((default_width, default_height))
strength = get_and_clamp_float(request.args, 'strength', 0.5, 1.0)
(model, provider, scheduler, prompt, negative_prompt, cfg, steps, height, (model, provider, scheduler, prompt, negative_prompt, cfg, steps, height,
width, seed, pipe) = pipeline_from_request(OnnxStableDiffusionInpaintPipeline) width, seed, pipe) = pipeline_from_request(OnnxStableDiffusionInpaintPipeline)
latents = get_latents_from_seed(seed, width, height)
rng = np.random.RandomState(seed) rng = np.random.RandomState(seed)
image = pipe( image = pipe(
prompt, prompt,
generator=rng, generator=rng,
guidance_scale=cfg, guidance_scale=cfg,
height=height,
image=source_image, image=source_image,
latents=latents,
mask_image=mask_image, mask_image=mask_image,
negative_prompt=negative_prompt, negative_prompt=negative_prompt,
num_inference_steps=steps, num_inference_steps=steps,
strength=strength, width=width,
).images[0] ).images[0]
(output_file, output_full) = make_output_path('inpaint', (prompt, cfg, steps, height, width, seed)) (output_file, output_full) = make_output_path('inpaint', (prompt, cfg, steps, height, width, seed))