Merge pull request #26 from provos/metal-support

google · Dec 22, 2024 · e5a4f3d · e5a4f3d
2 parents 4ef52b7 + d5315ee
commit e5a4f3d
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"]