Update .github/workflows/test_ipex.yml

Co-authored-by: Ella Charlaix <80481427+echarlaix@users.noreply.github.com>
huggingface · Jul 5, 2024 · b80511c · b80511c
1 parent 1b91ee6
commit b80511c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/test_ipex.yml b/.github/workflows/test_ipex.yml
@@ -35,8 +35,8 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        pip install torch==${{ matrix.torch-version }} --extra-index-url https://download.pytorch.org/whl/cpu
-        pip install intel_extension_for_pytorch==${{ matrix.intel_extension_for_pytorch-version }}
+        pip install torch==${{ matrix.ipex-version }} --extra-index-url https://download.pytorch.org/whl/cpu
+        pip install intel_extension_for_pytorch==${{ matrix.ipex-version }}
         pip install .[ipex,tests]
         pip install transformers==${{ matrix.transformers-version }}
     - name: Test with Pytest