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

Add requirements

Browse Source
This commit is contained in:
Woosuk Kwon 2024-04-01 03:17:43 +00:00
parent 86f073edd6
commit d148c2ef00
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
requirements-tpu.txt Normal file
View File

@ -0,0 +1,15 @@
psutil
ray >= 2.9
sentencepiece # Required for LLaMA tokenizer.
numpy
torch ~= 2.2.0
torch_xla[tpu] ~= 2.2.0
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