coreml

2025-04-03 00:35:19 -04:00
parent dd1fcd5be5
commit 543bc72ae3
2 changed files with 9 additions and 0 deletions
@@ -96,6 +96,14 @@ class OrtSession:
                        "precision": "FP32",
                        "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 _:
                    options = {}
            provider_options.append(options)