Skip to content

Commit

Permalink
Fix missing install candidate for ONNX Runtime on Apple Silicon
Browse files Browse the repository at this point in the history
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 pyannote#1505
  • Loading branch information
dr-duplo authored Oct 25, 2023
1 parent 0b45103 commit 65d6156
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 65d6156

Please sign in to comment.