Fix arm build (#13608)

blakeblackshear · Sep 7, 2024 · 17901fc · 17901fc
1 parent d6b16a7
commit 17901fc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/docker/main/requirements-wheels.txt b/docker/main/requirements-wheels.txt
@@ -30,7 +30,8 @@ ws4py == 0.5.*
 unidecode == 1.3.*
 # OpenVino & ONNX
 openvino == 2024.1.*
-onnxruntime-openvino == 1.18.*
+onnxruntime-openvino == 1.18.* ; platform_machine == 'x86_64'
+onnxruntime == 1.18.* ; platform_machine == 'aarch64'
 # Embeddings
 onnx_clip == 4.0.*
 chromadb == 0.5.0