Fix typo for auto_detect_ep function (#40)

mengniwang95 · web-flow · commit 3dd43d3d852d · 2024-11-18T14:53:17.000+08:00
## Type of Change

bug fix

---------

Signed-off-by: Wang, Mengni &lt;mengni.wang@intel.com&gt;
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

-Original file line number
+Diff line change
@@ @@ -1,3 +1,3 @@ @@
 optimum
 pytest
 -torch
 +torch < 2.5.0