requirements.txt 296 Bytes
Newer Older
1
2
ninja  # For faster builds.
psutil
3
ray >= 2.5.1
4
5
6
7
8
9
10
sentencepiece  # Required for LLaMA tokenizer.
numpy
torch >= 2.0.0
transformers >= 4.28.0  # Required for LLaMA.
xformers >= 0.0.19
fastapi
uvicorn
Keming's avatar
Keming committed
11
pydantic < 2  # Required for OpenAI server.
12
fschat  # Required for OpenAI ChatCompletion Endpoint.