From 00116ce4b469b6baa20a44a36de6d170af6e8150 Mon Sep 17 00:00:00 2001
From: ZiniuYu <ziniuyu@gmail.com>
Date: Tue, 21 Feb 2023 11:45:00 +0800
Subject: [PATCH] fix: onnx package conflict during setup

---
 server/setup.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/server/setup.py b/server/setup.py
index 7d850fabf..246323785 100644
--- 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'],