Skip to content

Commit d263bba

Browse files
authored
Rev ONNX and ORT deps (#242)
1 parent 163f071 commit d263bba

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

setup.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -31,12 +31,10 @@
3131
],
3232
install_requires=[
3333
"invoke>=2.0.0",
34-
# 1.16.2 causes a crash:
35-
# ImportError: DLL load failed while importing onnx_cpp2py_export
36-
"onnx>=1.11.0,<1.16.2",
34+
"onnx>=1.11.0",
3735
"onnxmltools==1.10.0",
3836
"onnxruntime >=1.10.1;platform_system=='Linux'",
39-
"onnxruntime-directml==1.19.0;platform_system=='Windows'",
37+
"onnxruntime-directml>=1.19.0;platform_system=='Windows'",
4038
"torch>=1.12.1",
4139
"pyyaml>=5.4",
4240
"typeguard>=2.3.13",
@@ -62,7 +60,6 @@
6260
"uvicorn[standard]",
6361
],
6462
"llm-oga-dml": [
65-
# "onnxruntime-directml==1.19.0",
6663
"onnxruntime-genai-directml==0.4.0",
6764
"tqdm",
6865
"torch>=2.0.0",

0 commit comments

Comments
 (0)