From 65d6156d0291f2c0eb5c9dba80e2c4d6781fd5e9 Mon Sep 17 00:00:00 2001
From: dr-duplo <monsieur.cm@gmx.de>
Date: Wed, 25 Oct 2023 14:43:41 +0200
Subject: [PATCH] Fix missing install candidate for ONNX Runtime on Apple
 Silicon

This patch alters the requirements to use another ONNX runtime package which provides pre-build wheels for for Apple Silicon in case of running on arm64/M1/M2 Mac.

Fixes #1505
---
 requirements.txt | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index 7829ada37..636359e40 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,7 +3,8 @@ einops >=0.6.0
 huggingface_hub >= 0.13.0
 lightning >= 2.0.1
 omegaconf >=2.1,<3.0
-onnxruntime-gpu >= 1.16.0
+onnxruntime-gpu >= 1.16.0; (sys_platform != "darwin" and platform_system != "Darwin") or platform_machine != "arm64"
+onnxruntime-silicon >= 1.16.0; (sys_platform == "darwin" or platform_system == "Darwin") and platform_machine == "arm64"
 pyannote.core >= 5.0.0
 pyannote.database >= 5.0.1
 pyannote.metrics >= 3.2