fix: onnx version

jina-ai · Feb 22, 2023 · bcd824c · bcd824c
1 parent ca1274a
commit bcd824c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/server/setup.py b/server/setup.py
@@ -54,7 +54,11 @@
             'onnx',
             'onnxmltools',
         ]
-        + (['onnxruntime-gpu'] if sys.platform != 'darwin' else ['onnxruntime']),
+        + (
+            ['onnxruntime-gpu<=1.13.1']
+            if sys.platform != 'darwin'
+            else ['onnxruntime<=1.13.1']
+        ),
         'tensorrt': [
             'nvidia-tensorrt==8.4.1.5',
         ],