apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

test multiple chained upscaling stages

This commit is contained in:
Sean Sube 2023-01-28 10:26:04 -06:00
parent ba3e450d43
commit 4188b019a1
2 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -63,12 +63,18 @@ def upscale_outpaint(
noise_image.save(base_join(ctx.output_path, 'last-noise.png'))
def outpaint(image: Image.Image, dims: Tuple[int, int, int]):
top, left, tile = dims
left, top, tile = dims
size = Size(*image.size)
mask = mask_image.crop((left, top, left + tile, top + tile))
if is_debug():
image.save(base_join(ctx.output_path, 'tile-source.png'))
mask.save(base_join(ctx.output_path, 'tile-mask.png'))
# TODO: must use inpainting model here
model = '../models/stable-diffusion-onnx-v1-inpainting'
pipe = load_pipeline(OnnxStableDiffusionInpaintPipeline,
params.model, params.provider, params.scheduler)
model, params.provider, params.scheduler)
latents = get_latents_from_seed(params.seed, size)
rng = np.random.RandomState(params.seed)
@ -87,7 +93,7 @@ def upscale_outpaint(
)
return result.images[0]
output = process_tiles(source_image, 256, 1, [outpaint])
output = process_tiles(source_image, 512, 1, [outpaint])
print('final output image size', output.size)
return output

8
api/onnx_web/serve.py
View File

@ -554,7 +554,13 @@ def chain():
}),
(persist_disk, StageParams(tile_size=8192), {
'output': output,
})
}),
(upscale_stable_diffusion, StageParams(tile_size=128,outscale=4), {
'upscale': UpscaleParams('stable-diffusion-x4-upscaler', params.provider, scale=4, outscale=4)
}),
(persist_disk, StageParams(tile_size=8192), {
'output': output,
}),
])
# build and run chain pipeline