Skip to content

Commit b2689e3

Browse files
authored
Use platform_system to install ort-directml on Windows (#239)
1 parent 50996bb commit b2689e3

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

setup.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,8 @@
3535
# ImportError: DLL load failed while importing onnx_cpp2py_export
3636
"onnx>=1.11.0,<1.16.2",
3737
"onnxmltools==1.10.0",
38-
"onnxruntime >=1.10.1",
38+
"onnxruntime >=1.10.1;platform_system=='Linux'",
39+
"onnxruntime-directml==1.19.0;platform_system=='Windows'",
3940
"torch>=1.12.1",
4041
"pyyaml>=5.4",
4142
"typeguard>=2.3.13",
@@ -61,7 +62,7 @@
6162
"uvicorn[standard]",
6263
],
6364
"llm-oga-dml": [
64-
"onnxruntime-directml==1.19.0",
65+
# "onnxruntime-directml==1.19.0",
6566
"onnxruntime-genai-directml==0.4.0",
6667
"tqdm",
6768
"torch>=2.0.0",

src/turnkeyml/version.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
__version__ = "4.0.4"
1+
__version__ = "4.0.5"

0 commit comments

Comments
 (0)