apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

add any option to platforms after checking providers

This commit is contained in:
Sean Sube 2023-02-11 16:01:55 -06:00
parent e4785b94e0
commit f45a0154e8
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

@ -344,9 +344,6 @@ def load_platforms(context: ServerContext):
providers = [] providers = []
providers.extend(get_available_providers()) providers.extend(get_available_providers())
if context.any_platform:
providers.append("any")
for potential in platform_providers: for potential in platform_providers:
if ( if (
platform_providers[potential] in providers platform_providers[potential] in providers
@ -368,6 +365,9 @@ def load_platforms(context: ServerContext):
DeviceParams(potential, platform_providers[potential]) DeviceParams(potential, platform_providers[potential])
) )
if context.any_platform:
available_platforms.append("any")
# make sure CPU is last on the list # make sure CPU is last on the list
def any_first_cpu_last(a: DeviceParams, b: DeviceParams): def any_first_cpu_last(a: DeviceParams, b: DeviceParams):
if a.device == b.device: if a.device == b.device: