File tree Expand file tree Collapse file tree 2 files changed +17
-2
lines changed
Expand file tree Collapse file tree 2 files changed +17
-2
lines changed Original file line number Diff line number Diff line change 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 .
Original file line number Diff line number Diff line change 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 .
You can’t perform that action at this time.
0 commit comments