diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py index e63f7282..5a630f3c 100644 --- a/api/onnx_web/serve.py +++ b/api/onnx_web/serve.py @@ -67,7 +67,10 @@ config_params = {} platform_providers = { 'amd': 'DmlExecutionProvider', 'cpu': 'CPUExecutionProvider', + 'cuda': 'CUDAExecutionProvider', + 'directml': 'DmlExecutionProvider', 'nvidia': 'CUDAExecutionProvider', + 'rocm': 'ROCmExecutionProvider', } pipeline_schedulers = { 'ddim': DDIMScheduler, diff --git a/gui/src/strings.ts b/gui/src/strings.ts index b7db329d..3069ee82 100644 --- a/gui/src/strings.ts +++ b/gui/src/strings.ts @@ -18,7 +18,9 @@ export const MODEL_LABELS = { export const PLATFORM_LABELS: Record = { amd: 'AMD GPU', cpu: 'CPU', - nvidia: 'Nvidia GPU', + directml: 'DirectML', + nvidia: 'CUDA', + rocm: 'ROCm', }; export const SCHEDULER_LABELS: Record = {