xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-10 00:45:01 +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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pyproject.toml
View File

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

4
requirements.txt
View File

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