15 lines
385 B
Plaintext
15 lines
385 B
Plaintext
ninja # For faster builds.
|
|
psutil
|
|
ray >= 2.5.1
|
|
pandas # Required for Ray data.
|
|
pyarrow # Required for Ray data.
|
|
sentencepiece # Required for LLaMA tokenizer.
|
|
numpy
|
|
torch == 2.1.2
|
|
transformers >= 4.36.0 # Required for Mixtral.
|
|
xformers == 0.0.23.post1 # Required for CUDA 12.1.
|
|
fastapi
|
|
uvicorn[standard]
|
|
pydantic == 1.10.13 # Required for OpenAI server.
|
|
aioprometheus[starlette]
|