Skip to content

Commit 255f4aa

Browse files
fix workflows
1 parent e48d92f commit 255f4aa

File tree

2 files changed

+17
-2
lines changed

2 files changed

+17
-2
lines changed

.github/workflows/test-paddle.yaml

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,8 +32,23 @@ jobs:
3232
- name: Install Python dependencies
3333
run: |
3434
python -m pip install --upgrade pip
35+
tf_ver=4.52
36+
npy_ver=2.2
37+
torch_ver=2.7
38+
if [ "${{ matrix.python-version }}" = "3.9" ]; then
39+
npy_ver=1.26
40+
tf_ver=4.40
41+
torch_ver=2.1
42+
elif [ "${{ matrix.python-version }}" = "3.10" ]; then
43+
torch_ver=2.3
44+
elif [ "${{ matrix.python-version }}" = "3.11" ]; then
45+
torch_ver=2.5
46+
elif [ "${{ matrix.python-version }}" = "3.12" ]; then
47+
torch_ver=2.6
48+
fi
49+
pip install torch==${torch_ver} --index-url https://download.pytorch.org/whl/cpu # transformers requires torch
3550
pip install paddlepaddle==3.0.0
36-
pip install pytest pytest-cov setuptools_scm safetensors transformers==4.52
51+
pip install vllm pytest pytest-cov setuptools_scm safetensors transformers==${tf_ver} numpy==${npy_ver}
3752
- name: Build Package
3853
run: |
3954
pip install .

.github/workflows/test-torch.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ jobs:
4747
torch_ver=2.6
4848
fi
4949
pip install torch==${torch_ver} --index-url https://download.pytorch.org/whl/cpu
50-
pip install pytest pytest-cov setuptools_scm safetensors transformers==${tf_ver} numpy==${npy_ver}
50+
pip install vllm pytest pytest-cov setuptools_scm safetensors transformers==${tf_ver} numpy==${npy_ver}
5151
- name: Build package
5252
run: |
5353
pip install .

0 commit comments

Comments
 (0)