read chain params per stage

2023-09-10 21:23:16 -05:00 · 2023-09-10 21:23:16 -05:00 · d0080c476d
parent 441a47a885
commit d0080c476d
1 changed files with 1 additions and 0 deletions
--- a/api/onnx_web/server/api.py
+++ b/api/onnx_web/server/api.py
@ -395,6 +395,7 @@ def chain(server: ServerContext, pool: DevicePoolExecutor):
        kwargs: Dict[str, Any] = stage_data.get("params", {})
        logger.info("request stage: %s, %s", stage_class.__name__, kwargs)

+        # TODO: combine base params with stage params
        _device, params, size = pipeline_from_request(server, data=kwargs)
        replace_wildcards(params, get_wildcard_data())