Skip to content

Commit e53954e

Browse files
committed
run in docker
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
1 parent 52678e7 commit e53954e

File tree

1 file changed

+5
-19
lines changed

1 file changed

+5
-19
lines changed

.github/workflows/test_ipex.yml

+5-19
Original file line numberDiff line numberDiff line change
@@ -18,38 +18,24 @@ jobs:
1818
strategy:
1919
fail-fast: false
2020
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.*"]
2323

24-
runs-on: ubuntu-22.04
24+
runs-on: ubuntu-latest
25+
container: intel/intel-extension-for-pytorch:2.6.0-pip-base
2526

2627
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-
3528
- name: Install dependencies
3629
run: |
3730
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 }}
4032
4133
- name: Install bitsandbytes
4234
run: |
4335
git clone --branch multi-backend-refactor https://github.com/bitsandbytes-foundation/bitsandbytes.git
4436
cd bitsandbytes
4537
pip install .
4638
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-
5339
- name: Assert versions
5440
run: |
5541
python -c "import torch; print(torch.__version__); assert torch.__version__.startswith('${{ matrix.torch-version }}'.replace('.*', ''))"

0 commit comments

Comments
 (0)