apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): respect model selection when running inpaint/outpaint stages

This commit is contained in:
Sean Sube 2023-02-02 18:51:54 -06:00
parent 406e0f0cf9
commit a1298f1565
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/chain/blend_inpaint.py
View File

@ -78,10 +78,8 @@ def blend_inpaint(
save_image(ctx, 'tile-source.png', image)
save_image(ctx, 'tile-mask.png', mask)
# TODO: must use inpainting model here
model = '../models/stable-diffusion-onnx-v1-inpainting'
pipe = load_pipeline(OnnxStableDiffusionInpaintPipeline,
model, params.provider, params.scheduler)
params.model, params.provider, params.scheduler)
latents = get_latents_from_seed(params.seed, size)
rng = np.random.RandomState(params.seed)

4
api/onnx_web/chain/upscale_outpaint.py
View File

@ -86,10 +86,8 @@ def upscale_outpaint(
save_image(ctx, 'tile-source.png', image)
save_image(ctx, 'tile-mask.png', mask)
# TODO: must use inpainting model here
model = '../models/stable-diffusion-onnx-v1-inpainting'
pipe = load_pipeline(OnnxStableDiffusionInpaintPipeline,
model, params.provider, params.scheduler)
params.model, params.provider, params.scheduler)
latents = get_tile_latents(full_latents, dims)
rng = np.random.RandomState(params.seed)