vllm/requirements-rocm.txt

19 lines
414 B
Plaintext
Raw Normal View History

2024-03-19 06:38:33 +08:00
cmake>=3.21
ninja # For faster builds.
typing-extensions>=4.8.0
starlette
2024-03-29 13:16:12 +08:00
requests
py-cpuinfo
psutil
ray == 2.9.3
sentencepiece # Required for LLaMA tokenizer.
numpy
tokenizers>=0.15.0
transformers >= 4.39.1 # Required for StarCoder2 & Llava.
fastapi
uvicorn[standard]
2024-01-22 08:05:56 +08:00
pydantic >= 2.0 # Required for OpenAI server.
prometheus_client >= 0.18.0
outlines == 0.0.34
tiktoken == 0.6.0 # Required for DBRX tokenizer