From 38c8a8a4139a918c90f20ad5aced04363b6cab42 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 11 Feb 2023 16:06:14 -0600 Subject: [PATCH] make any a real device --- api/onnx_web/serve.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py index 2f092d2a..c82d4694 100644 --- a/api/onnx_web/serve.py +++ b/api/onnx_web/serve.py @@ -341,8 +341,7 @@ def load_params(context: ServerContext): def load_platforms(context: ServerContext): global available_platforms - providers = [] - providers.extend(get_available_providers()) + providers = list(get_available_providers()) for potential in platform_providers: if ( @@ -366,7 +365,8 @@ def load_platforms(context: ServerContext): ) if context.any_platform: - available_platforms.append("any") + # the platform should be ignored when the job is scheduled, but set to CPU just in case + available_platforms.append(DeviceParams("any", platform_providers["cpu"])) # make sure CPU is last on the list def any_first_cpu_last(a: DeviceParams, b: DeviceParams):