diff --git a/docker/main/requirements-wheels.txt b/docker/main/requirements-wheels.txt
index 756bf18b5..a81f6ec93 100644
--- 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