From d5315ee15d466e801920c09d704af1731a7b2fc4 Mon Sep 17 00:00:00 2001 From: provos Date: Fri, 20 Dec 2024 18:06:15 -0500 Subject: [PATCH] feat: add support for CoreMLExecutionProvider in ONNX backend --- unisim/backend/onnx.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/unisim/backend/onnx.py b/unisim/backend/onnx.py index 036daf0..189798b 100644 --- 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"]