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