feat: add ROCm provider to list (#10)
This commit is contained in:
parent
137a5a84cc
commit
3bcd7a8156
|
@ -67,7 +67,10 @@ config_params = {}
|
|||
platform_providers = {
|
||||
'amd': 'DmlExecutionProvider',
|
||||
'cpu': 'CPUExecutionProvider',
|
||||
'cuda': 'CUDAExecutionProvider',
|
||||
'directml': 'DmlExecutionProvider',
|
||||
'nvidia': 'CUDAExecutionProvider',
|
||||
'rocm': 'ROCmExecutionProvider',
|
||||
}
|
||||
pipeline_schedulers = {
|
||||
'ddim': DDIMScheduler,
|
||||
|
|
|
@ -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> = {
|
||||
|
|
Loading…
Reference in New Issue