Fix typo for auto_detect_ep function (#40)

## Type of Change bug fix --------- Signed-off-by: Wang, Mengni <[email protected]>
onnx · Nov 18, 2024 · 3dd43d3 · 3dd43d3
1 parent aabbf96
commit 3dd43d3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/onnx_neural_compressor/utility.py b/onnx_neural_compressor/utility.py
@@ -306,7 +306,7 @@ def auto_detect_ep():
     if "DnnlExecutionProvider" in eps:
         return "DnnlExecutionProvider"
     elif "DmlExecutionProvider" in eps:
-        return "DnnlExecutionProvider"
+        return "DmlExecutionProvider"
     elif "CUDAExecutionProvider" in eps:
         return "CUDAExecutionProvider"
     else:

diff --git a/test/requirements.txt b/test/requirements.txt
@@ -1,3 +1,3 @@
 optimum
 pytest
-torch
+torch < 2.5.0