helpy4/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich synced 2025-11-07 17:27:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

coreml

Browse source
This commit is contained in:
mertalev 2025-04-03 00:35:19 -04:00
parent 46869f664d
commit 058a10d0c6
No known key found for this signature in database
GPG key ID: DF6ABC77AAD98C95
2 changed files with 9 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
machine-learning/immich_ml/models/constants.py
View file

@ -79,6 +79,7 @@ SUPPORTED_PROVIDERS = [
"CUDAExecutionProvider", "CUDAExecutionProvider",
"ROCMExecutionProvider", "ROCMExecutionProvider",
"OpenVINOExecutionProvider", "OpenVINOExecutionProvider",
"CoreMLExecutionProvider",
"CPUExecutionProvider", "CPUExecutionProvider",
] ]

8
machine-learning/immich_ml/sessions/ort.py
View file

@ -96,6 +96,14 @@ class OrtSession:
"precision": "FP32", "precision": "FP32",
"cache_dir": (self.model_path.parent / "openvino").as_posix(), "cache_dir": (self.model_path.parent / "openvino").as_posix(),
} }
case "CoreMLExecutionProvider":
options = {
"ModelFormat": "MLProgram",
"MLComputeUnits": "ALL",
"SpecializationStrategy": "FastPrediction",
"AllowLowPrecisionAccumulationOnGPU": "1",
"ModelCacheDirectory": (self.model_path.parent / "coreml").as_posix(),
}
case _: case _:
options = {} options = {}
provider_options.append(options) provider_options.append(options)