apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

make any a real device

This commit is contained in:
Sean Sube 2023-02-11 16:06:14 -06:00
parent f45a0154e8
commit 38c8a8a413
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/onnx_web/serve.py
View File

@ -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):