Rev ONNX and ORT deps (#242)

onnx · Nov 25, 2024 · d263bba · d263bba
1 parent 163f071
commit d263bba
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/setup.py b/setup.py
@@ -31,12 +31,10 @@
     ],
     install_requires=[
         "invoke>=2.0.0",
-        # 1.16.2 causes a crash:
-        #   ImportError: DLL load failed while importing onnx_cpp2py_export
-        "onnx>=1.11.0,<1.16.2",
+        "onnx>=1.11.0",
         "onnxmltools==1.10.0",
         "onnxruntime >=1.10.1;platform_system=='Linux'",
-        "onnxruntime-directml==1.19.0;platform_system=='Windows'",
+        "onnxruntime-directml>=1.19.0;platform_system=='Windows'",
         "torch>=1.12.1",
         "pyyaml>=5.4",
         "typeguard>=2.3.13",
@@ -62,7 +60,6 @@
             "uvicorn[standard]",
         ],
         "llm-oga-dml": [
-            # "onnxruntime-directml==1.19.0",
             "onnxruntime-genai-directml==0.4.0",
             "tqdm",
             "torch>=2.0.0",