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

Fix requirements

Browse Source
This commit is contained in:
Woosuk Kwon 2024-04-10 07:52:45 +00:00
parent 60ff6b8c5c
commit 0d6402ddfd
1 changed files with 2 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
requirements-tpu.txt
View File

@ -1,16 +1,2 @@
psutil # Common dependencies
ray >= 2.9 -r requirements-common.txt
sentencepiece # Required for LLaMA tokenizer.
numpy
torch ~= 2.2.0
torch_xla[tpu] ~= 2.2.0
jax[tpu] # Required for Pallas kernels.
requests
py-cpuinfo
transformers >= 4.39.1 # Required for StarCoder2 & Llava.
fastapi
uvicorn[standard]
pydantic >= 2.0 # Required for OpenAI server.
prometheus_client >= 0.18.0
outlines == 0.0.34
tiktoken == 0.6.0 # Required for DBRX tokenizer