frozenleaves/vllm
1
0
Fork 0
mirror of https://github.com/vllm-project/vllm.git synced 2025-10-20 14:53:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

lock torch version to 2.0.1 (#1290)

Browse Source
This commit is contained in:
yanxiyue
2023-10-11 00:21:57 +08:00
committed by GitHub
parent b95ee898fe
commit 6a6119554c
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pyproject.toml
View File

@ -3,7 +3,7 @@ requires = [
"ninja",
"packaging",
"setuptools",
"torch >= 2.0.0",
"torch == 2.0.1",
"wheel",
]
build-backend = "setuptools.build_meta"

4
requirements.txt
View File

@ -5,9 +5,9 @@ pandas # Required for Ray data.
pyarrow # Required for Ray data.
sentencepiece # Required for LLaMA tokenizer.
numpy
torch >= 2.0.0
torch == 2.0.1
transformers >= 4.33.1 # Required for Code Llama.
xformers >= 0.0.22
xformers == 0.0.22
fastapi
uvicorn[standard]
pydantic < 2 # Required for OpenAI server.