vllm/requirements-rocm.txt
2024-03-24 06:32:45 -07:00

16 lines
344 B
Plaintext

cmake>=3.21
ninja # For faster builds.
typing-extensions>=4.8.0
starlette
psutil
ray >= 2.9
sentencepiece # Required for LLaMA tokenizer.
numpy
tokenizers>=0.15.0
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