From 3bcd7a8156328dbc0087efc12450abf22df443fd Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Thu, 19 Jan 2023 17:37:26 -0600
Subject: [PATCH] feat: add ROCm provider to list (#10)

---
 api/onnx_web/serve.py | 3 +++
 gui/src/strings.ts    | 4 +++-
 2 files changed, 6 insertions(+), 1 deletion(-)

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<string, string> = {
   amd: 'AMD GPU',
   cpu: 'CPU',
-  nvidia: 'Nvidia GPU',
+  directml: 'DirectML',
+  nvidia: 'CUDA',
+  rocm: 'ROCm',
 };
 
 export const SCHEDULER_LABELS: Record<string, string> = {