xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-10 05:55:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

Remove einops from requirements (#2049)

Browse Source
This commit is contained in:
Woosuk Kwon 2023-12-12 09:34:09 -08:00 committed by GitHub
parent cb3f30c600
commit 7e1b21daac
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
requirements-rocm.txt
View File

@ -9,7 +9,6 @@ sentencepiece # Required for LLaMA tokenizer.
numpy numpy
tokenizers>=0.15.0 tokenizers>=0.15.0
huggingface_hub<0.18,>=0.16.4 huggingface_hub<0.18,>=0.16.4
einops # Required for phi-1_5
transformers >= 4.36.0 # Required for Mixtral. transformers >= 4.36.0 # Required for Mixtral.
fastapi fastapi
uvicorn[standard] uvicorn[standard]

1
requirements.txt
View File

@ -5,7 +5,6 @@ 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
einops # Required for phi-1_5
torch >= 2.1.1 torch >= 2.1.1
transformers >= 4.36.0 # Required for Mixtral. transformers >= 4.36.0 # Required for Mixtral.
xformers >= 0.0.23 # Required for CUDA 12.1. xformers >= 0.0.23 # Required for CUDA 12.1.