cjones/immich
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(ml): introduce support of onnxruntime-rocm for AMD GPU

Browse Source
This commit is contained in:
Mehdi GHESH
2024-07-13 00:40:29 +02:00
committed by mertalev
parent 3f4bbab4eb
commit fe26ccd1b7
14 changed files with 275 additions and 78 deletions
Download Patch File Download Diff File Expand all files Collapse all files
machine-learning/app/models/constants.py
+1 -1
View File
@@ -63,7 +63,7 @@ _INSIGHTFACE_MODELS = {
}
SUPPORTED_PROVIDERS = ["CUDAExecutionProvider", "OpenVINOExecutionProvider", "CPUExecutionProvider"]
SUPPORTED_PROVIDERS = ["CUDAExecutionProvider", "ROCMExecutionProvider", "OpenVINOExecutionProvider", "CPUExecutionProvider"]
def get_model_source(model_name: str) -> ModelSource | None: