Skip to content

Commit 4bee0c7

Browse files
authoredApr 25, 2024··
Pin Pytorch 2.2 in IPEX Tests (#688)
* pin torch * pin in ipex test only * fix inc test
1 parent 7a4464d commit 4bee0c7

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed
 

‎.github/workflows/test_inc.yml

+1
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@ jobs:
3232
python -m pip install --upgrade pip
3333
pip install cmake
3434
pip install py-cpuinfo
35+
pip install torch==2.2 torchaudio torchvision --extra-index-url https://download.pytorch.org/whl/cpu
3536
pip install .[neural-compressor,diffusers,tests]
3637
pip install intel-extension-for-transformers
3738
pip install peft

‎.github/workflows/test_ipex.yml

+1
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ jobs:
3030
- name: Install dependencies
3131
run: |
3232
python -m pip install --upgrade pip
33+
pip install torch==2.2 torchaudio torchvision --extra-index-url https://download.pytorch.org/whl/cpu
3334
pip install .[ipex,tests]
3435
- name: Test with Pytest
3536
run: |

0 commit comments

Comments
 (0)
Please sign in to comment.