diff --git a/docker/main/requirements-wheels.txt b/docker/main/requirements-wheels.txt
index 224f3d20a..756bf18b5 100644
--- a/docker/main/requirements-wheels.txt
+++ b/docker/main/requirements-wheels.txt
@@ -28,8 +28,9 @@ norfair == 2.2.*
 setproctitle == 1.3.*
 ws4py == 0.5.*
 unidecode == 1.3.*
-onnxruntime == 1.18.*
+# OpenVino & ONNX
 openvino == 2024.1.*
+onnxruntime-openvino == 1.18.*
 # Embeddings
 onnx_clip == 4.0.*
 chromadb == 0.5.0
@@ -39,4 +40,4 @@ ollama == 0.2.*
 openai == 1.30.*
 # push notifications
 py-vapid == 1.9.*
-pywebpush == 2.0.*
\ No newline at end of file
+pywebpush == 2.0.*