feat: add support for CoreMLExecutionProvider in ONNX backend

google · Dec 20, 2024 · d5315ee · d5315ee
1 parent 4ef52b7
commit d5315ee
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/unisim/backend/onnx.py b/unisim/backend/onnx.py
@@ -32,6 +32,8 @@ def cosine_similarity(query_embeddings: BatchEmbeddings, index_embeddings: Batch
 avail_providers = get_available_providers()
 if "CUDAExecutionProvider" in avail_providers:
     _providers = ["CUDAExecutionProvider", "CPUExecutionProvider"]
+elif "CoreMLExecutionProvider" in avail_providers:
+    _providers = ["CoreMLExecutionProvider", "CPUExecutionProvider"]
 else:
     _providers = ["CPUExecutionProvider"]