apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

wire up per stage params

This commit is contained in:
Sean Sube 2023-09-10 22:21:48 -05:00
parent 89de05b7c5
commit fb7a0996a8
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
api/onnx_web/chain/base.py
View File

@ -123,6 +123,11 @@ class ChainPipeline:
kwargs.keys(),
)
per_stage_params = params
if "params" in kwargs:
per_stage_params = kwargs["params"]
kwargs.pop("params")
# the stage must be split and tiled if any image is larger than the selected/max tile size
must_tile = any(
[
@ -159,7 +164,7 @@ class ChainPipeline:
worker,
server,
stage_params,
kwargs["params"] if "params" in kwargs else params,
per_stage_params,
[source_tile],
tile_mask=tile_mask,
callback=callback,
@ -201,7 +206,7 @@ class ChainPipeline:
worker,
server,
stage_params,
params,
per_stage_params,
stage_sources,
callback=callback,
**kwargs,