requirements.txt

ninja  # For faster builds.
psutil
ray >= 2.5.1
sentencepiece  # Required for LLaMA tokenizer.
numpy
torch >= 2.0.0
transformers >= 4.31.0  # Required for LLaMA-2.
xformers >= 0.0.19
fastapi
uvicorn
pydantic < 2  # Required for OpenAI server.
fschat  # Required for OpenAI ChatCompletion Endpoint.