diff --git a/api/onnx_web/server/params.py b/api/onnx_web/server/params.py index a9d8aef9..2f505432 100644 --- a/api/onnx_web/server/params.py +++ b/api/onnx_web/server/params.py @@ -46,7 +46,7 @@ def pipeline_from_request( # diffusion model model = get_not_empty(request.args, "model", get_config_value("model")) model_path = get_model_path(server, model) - control = get_from_list(request.args, "control", get_network_models()) + control = get_from_list(request.args, "control", [m.name for m in get_network_models()]) # pipeline stuff pipeline = get_from_list( diff --git a/api/onnx_web/utils.py b/api/onnx_web/utils.py index 1e1f875b..23980fdd 100644 --- a/api/onnx_web/utils.py +++ b/api/onnx_web/utils.py @@ -45,7 +45,7 @@ def get_from_list( if selected in values: return selected - logger.warn("invalid selection: %s", selected) + logger.warn("invalid selection %s, options: %s", selected, values) if len(values) > 0: return values[0]