File tree 1 file changed +3
-3
lines changed
1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 32
32
python -m pip install --upgrade pip
33
33
pip install cmake
34
34
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
36
36
pip install .[neural-compressor,diffusers,tests]
37
37
pip install intel-extension-for-transformers
38
38
pip install peft
43
43
- name : Test IPEX
44
44
run : |
45
45
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
48
48
pytest tests/neural_compressor/test_ipex.py
49
49
You can’t perform that action at this time.
0 commit comments