requirements.txt 323 Bytes
Newer Older
1
2
ninja  # For faster builds.
psutil
3
ray >= 2.5.1
4
pandas  # Required for Ray data.
5
pyarrow  # Required for Ray data.
6
7
8
sentencepiece  # Required for LLaMA tokenizer.
numpy
torch >= 2.0.0
9
transformers >= 4.33.1  # Required for Code Llama.
10
xformers >= 0.0.22
11
fastapi
12
uvicorn[standard]
Keming's avatar
Keming committed
13
pydantic < 2  # Required for OpenAI server.