fix: onnx package conflict during setup

jina-ai · Feb 21, 2023 · 00116ce · 00116ce
1 parent dabbe8b
commit 00116ce
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/server/setup.py b/server/setup.py
@@ -51,11 +51,10 @@
     ],
     extras_require={
         'onnx': [
-            'onnxruntime',
             'onnx',
             'onnxmltools',
         ]
-        + (['onnxruntime-gpu>=1.8.0'] if sys.platform != 'darwin' else []),
+        + (['onnxruntime-gpu>=1.8.0'] if sys.platform != 'darwin' else ['onnxruntime']),
         'tensorrt': ['nvidia-tensorrt'],
         'transformers': ['transformers>=4.16.2'],
         'search': ['annlite>=0.3.10'],