Skip to content

Commit c98f2b0

Browse files
update deps (#2368)
v2 release
1 parent f1f6ae4 commit c98f2b0

File tree

1 file changed

+3
-6
lines changed

1 file changed

+3
-6
lines changed

setup.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -43,12 +43,6 @@
4343
BENCHMARK_REQUIRE = ["optuna", "tqdm", "scikit-learn", "seqeval", "torchvision", "evaluate>=0.2.0"]
4444

4545
EXTRAS_REQUIRE = {
46-
###########################################################################
47-
# until optimum-onnx is released on PyPI
48-
"onnx": "optimum-onnx @ git+https://github.com/huggingface/optimum-onnx.git",
49-
"onnxruntime": "optimum-onnx[onnxruntime] @ git+https://github.com/huggingface/optimum-onnx.git",
50-
"onnxruntime-gpu": "optimum-onnx[onnxruntime-gpu] @ git+https://github.com/huggingface/optimum-onnx.git",
51-
###########################################################################
5246
"amd": "optimum-amd",
5347
"furiosa": "optimum-furiosa",
5448
"graphcore": "optimum-graphcore",
@@ -57,6 +51,9 @@
5751
"ipex": "optimum-intel[ipex]>=1.23.0",
5852
"nncf": "optimum-intel[nncf]>=1.23.0",
5953
"neural-compressor": "optimum-intel[neural-compressor]>=1.23.0",
54+
"onnx": "optimum-onnx",
55+
"onnxruntime": "optimum-onnx[onnxruntime]",
56+
"onnxruntime-gpu": "optimum-onnx[onnxruntime-gpu]",
6057
"openvino": "optimum-intel[openvino]>=1.23.0",
6158
"quanto": "optimum-quanto>=0.2.4",
6259
###########################################################################

0 commit comments

Comments
 (0)