apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

pass size and initial image to test pipeline

This commit is contained in:
Sean Sube 2023-01-28 08:40:29 -06:00
parent 9f76226ba6
commit 658e12266f
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/onnx_web/serve.py
View File

@ -545,14 +545,16 @@ def chain():
params, size = pipeline_from_request()
example = ChainPipeline(stages=[
(generate_txt2img, StageParams(), None),
(generate_txt2img, StageParams(), {
'size': size,
}),
(upscale_outpaint, StageParams(outscale=4), {
'expand': Border(256, 256, 256, 256),
}),
])
output = make_output_name('chain', params, size)
executor.submit_stored(output, example, context, params, None)
executor.submit_stored(output, example, context, params, Image.new('RGB', (1, 1)))
# parse body as json, list of stages
# build and run chain pipeline