From 15c651b31a1cceca45635d156e3c7aff2b6c081e Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 5 Feb 2023 13:44:13 -0600 Subject: [PATCH] fix(api): use first available platform if requested one does not exist --- api/onnx_web/serve.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py index 28680c88..1ccdd8e8 100644 --- a/api/onnx_web/serve.py +++ b/api/onnx_web/serve.py @@ -168,7 +168,8 @@ def pipeline_from_request() -> Tuple[DeviceParams, ImageParams, Size]: device = available_platforms[0] if device is None: - raise Exception("unknown device") + logger.warn('unknown platform: %s', device_name) + device = available_platforms[0] # pipeline stuff model = get_not_empty(request.args, "model", get_config_value("model"))