20231088/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[CI/CD] Upgrade PyTorch version to v2.1.1 (#2045)

Browse Source
This commit is contained in:
Woosuk Kwon 2023-12-11 17:48:11 -08:00 committed by GitHub
parent 31d2ab4aff
commit f3e024bece
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/publish.yml vendored
View File

@ -49,7 +49,7 @@ jobs:
matrix:
os: ['ubuntu-20.04']
python-version: ['3.8', '3.9', '3.10', '3.11']
pytorch-version: ['2.1.0']
pytorch-version: ['2.1.1']
cuda-version: ['11.8', '12.1']
steps:

2
pyproject.toml
View File

@ -4,7 +4,7 @@ requires = [
"ninja",
"packaging",
"setuptools >= 49.4.0",
"torch >= 2.1.0",
"torch >= 2.1.1",
"wheel",
]
build-backend = "setuptools.build_meta"

4
requirements.txt
View File

@ -6,9 +6,9 @@ pyarrow # Required for Ray data.
sentencepiece # Required for LLaMA tokenizer.
numpy
einops # Required for phi-1_5
torch >= 2.1.0
torch >= 2.1.1
transformers >= 4.34.0 # Required for Mistral.
xformers >= 0.0.22.post7 # Required for CUDA 12.1.
xformers >= 0.0.23 # Required for CUDA 12.1.
fastapi
uvicorn[standard]
pydantic == 1.10.13 # Required for OpenAI server.