From b2689e3c9d99d296a825c1c86d7f7a6b06558556 Mon Sep 17 00:00:00 2001
From: Jeremy Fowers <80718789+jeremyfowers@users.noreply.github.com>
Date: Wed, 20 Nov 2024 14:58:27 -0500
Subject: [PATCH] Use platform_system to install ort-directml on Windows (#239)

---
 setup.py                 | 5 +++--
 src/turnkeyml/version.py | 2 +-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/setup.py b/setup.py
index 5933accc..142655a5 100644
--- a/setup.py
+++ b/setup.py
@@ -35,7 +35,8 @@
         #   ImportError: DLL load failed while importing onnx_cpp2py_export
         "onnx>=1.11.0,<1.16.2",
         "onnxmltools==1.10.0",
-        "onnxruntime >=1.10.1",
+        "onnxruntime >=1.10.1;platform_system=='Linux'",
+        "onnxruntime-directml==1.19.0;platform_system=='Windows'",
         "torch>=1.12.1",
         "pyyaml>=5.4",
         "typeguard>=2.3.13",
@@ -61,7 +62,7 @@
             "uvicorn[standard]",
         ],
         "llm-oga-dml": [
-            "onnxruntime-directml==1.19.0",
+            # "onnxruntime-directml==1.19.0",
             "onnxruntime-genai-directml==0.4.0",
             "tqdm",
             "torch>=2.0.0",
diff --git a/src/turnkeyml/version.py b/src/turnkeyml/version.py
index 4b56dfc5..e2a8e2c0 100644
--- a/src/turnkeyml/version.py
+++ b/src/turnkeyml/version.py
@@ -1 +1 @@
-__version__ = "4.0.4"
+__version__ = "4.0.5"