apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

feat: add ROCm provider to list (#10)

This commit is contained in:
Sean Sube 2023-01-19 17:37:26 -06:00
parent 137a5a84cc
commit 3bcd7a8156
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/onnx_web/serve.py
View File

@ -67,7 +67,10 @@ config_params = {}
platform_providers = { platform_providers = {
'amd': 'DmlExecutionProvider', 'amd': 'DmlExecutionProvider',
'cpu': 'CPUExecutionProvider', 'cpu': 'CPUExecutionProvider',
'cuda': 'CUDAExecutionProvider',
'directml': 'DmlExecutionProvider',
'nvidia': 'CUDAExecutionProvider', 'nvidia': 'CUDAExecutionProvider',
'rocm': 'ROCmExecutionProvider',
} }
pipeline_schedulers = { pipeline_schedulers = {
'ddim': DDIMScheduler, 'ddim': DDIMScheduler,

4
gui/src/strings.ts
View File

@ -18,7 +18,9 @@ export const MODEL_LABELS = {
export const PLATFORM_LABELS: Record<string, string> = { export const PLATFORM_LABELS: Record<string, string> = {
amd: 'AMD GPU', amd: 'AMD GPU',
cpu: 'CPU', cpu: 'CPU',
nvidia: 'Nvidia GPU', directml: 'DirectML',
nvidia: 'CUDA',
rocm: 'ROCm',
}; };
export const SCHEDULER_LABELS: Record<string, string> = { export const SCHEDULER_LABELS: Record<string, string> = {