Set optimum-intel to 1.16.1 (#1590)

changwangss · web-flow · commit 2f7feb730eef · 2024-06-04T17:07:53.000+08:00
Signed-off-by: Wang, Chang &lt;chang1.wang@intel.com&gt;
diff --git a/.github/workflows/unit-test-optimize.yml b/.github/workflows/unit-test-optimize.yml
@@ -43,6 +43,7 @@ jobs:
             test_name: "PR-test"
           - test_branch: "main"
             test_name: "baseline"
+      fail-fast: true
     name: optimize-unit-test-${{ matrix.test_name }}
     steps:
       - name: Docker Clean Up
diff --git a/tests/requirements.txt b/tests/requirements.txt
@@ -16,7 +16,7 @@ mlflow
 nlpaug==1.1.9
 onnx
 onnxruntime
-optimum
+optimum-intel==1.16.1
 peft==0.6.2
 py-cpuinfo
 sacremoses
