update test

Signed-off-by: chensuyue <[email protected]>
intel · Jul 28, 2024 · f21d17d · f21d17d
1 parent 9eef63b
commit f21d17d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/.azure-pipelines/scripts/install_nc.sh b/.azure-pipelines/scripts/install_nc.sh
@@ -6,6 +6,9 @@ if [[ $1 = *"3x_pt"* ]]; then
     python -m pip install --no-cache-dir -r requirements_pt.txt
     python setup.py pt bdist_wheel
     pip install --no-deps dist/neural_compressor*.whl --force-reinstall
+    if [[ $1 != *"3x_pt_fp8"* ]]; then
+        pip install torch --index-url https://download.pytorch.org/whl/cpu
+    fi
 elif [[ $1 = *"3x_tf"* ]]; then
     python -m pip install --no-cache-dir -r requirements_tf.txt
     python setup.py tf bdist_wheel