Skip to content

Commit 82782c4

Browse files
committedMay 14, 2024
set latest torch version
1 parent 7b4c8d9 commit 82782c4

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed
 

‎.github/workflows/test_inc.yml

+3-3
Original file line numberDiff line numberDiff line change
@@ -32,7 +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
35+
pip install torch torchaudio torchvision --extra-index-url https://download.pytorch.org/whl/cpu
3636
pip install .[neural-compressor,diffusers,tests]
3737
pip install intel-extension-for-transformers
3838
pip install peft
@@ -43,7 +43,7 @@ jobs:
4343
- name: Test IPEX
4444
run: |
4545
pip uninstall -y intel-extension-for-transformers
46-
pip install torch==2.1.0 torchaudio==2.1.0 torchvision==0.16 --extra-index-url https://download.pytorch.org/whl/cpu
47-
pip install intel-extension-for-pytorch==2.1.100
46+
pip install torch==2.3.0 torchaudio==2.3.0 torchvision==0.18 --extra-index-url https://download.pytorch.org/whl/cpu
47+
pip install intel-extension-for-pytorch==2.3.0
4848
pytest tests/neural_compressor/test_ipex.py
4949

0 commit comments

Comments
 (0)