fix(api): respect model selection when running inpaint/outpaint stages
This commit is contained in:
parent
406e0f0cf9
commit
a1298f1565
|
@ -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)
|
||||
|
|
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue