Commit e53954e 1 parent 52678e7 commit e53954e Copy full SHA for e53954e
File tree 1 file changed +5
-19
lines changed
1 file changed +5
-19
lines changed Original file line number Diff line number Diff line change @@ -18,38 +18,24 @@ jobs:
18
18
strategy :
19
19
fail-fast : false
20
20
matrix :
21
- transformers -version : ["4.47.0", "4.47.1 "]
22
- torch -version : ["2.4.0", "2.5 .*"]
21
+ torch -version : ["2.6.0 "]
22
+ transformers -version : ["4.47 .*"]
23
23
24
- runs-on : ubuntu-22.04
24
+ runs-on : ubuntu-latest
25
+ container : intel/intel-extension-for-pytorch:2.6.0-pip-base
25
26
26
27
steps :
27
- - name : Checkout code
28
- uses : actions/checkout@v4
29
-
30
- - name : Setup Python
31
- uses : actions/setup-python@v5
32
- with :
33
- python-version : " 3.10"
34
-
35
28
- name : Install dependencies
36
29
run : |
37
30
pip install --upgrade pip
38
- pip install torch==${{ matrix.torch-version }} torchaudio torchvision --extra-index-url https://download.pytorch.org/whl/cpu
39
- pip install .[ipex,tests] transformers[testing]==${{ matrix.transformers-version }} intel_extension_for_pytorch==${{ matrix.torch-version }}
31
+ pip install .[ipex,tests] transformers[testing]==${{ matrix.transformers-version }}
40
32
41
33
- name : Install bitsandbytes
42
34
run : |
43
35
git clone --branch multi-backend-refactor https://github.com/bitsandbytes-foundation/bitsandbytes.git
44
36
cd bitsandbytes
45
37
pip install .
46
38
47
- - name : Install autoawq
48
- run : |
49
- git clone --branch setup https://github.com/jiqing-feng/AutoAWQ.git
50
- cd AutoAWQ
51
- pip install .
52
-
53
39
- name : Assert versions
54
40
run : |
55
41
python -c "import torch; print(torch.__version__); assert torch.__version__.startswith('${{ matrix.torch-version }}'.replace('.*', ''))"
You can’t perform that action at this time.
0 commit comments