2023-05-08 07:30:43 +08:00
|
|
|
ninja # For faster builds.
|
|
|
|
|
psutil
|
2023-07-20 13:49:31 +08:00
|
|
|
ray >= 2.5.1
|
2023-09-15 08:31:38 +08:00
|
|
|
pandas # Required for Ray data.
|
2023-09-19 13:36:17 +08:00
|
|
|
pyarrow # Required for Ray data.
|
2023-05-08 07:30:43 +08:00
|
|
|
sentencepiece # Required for LLaMA tokenizer.
|
|
|
|
|
numpy
|
2023-12-12 09:48:11 +08:00
|
|
|
torch >= 2.1.1
|
2023-12-12 10:39:14 +08:00
|
|
|
transformers >= 4.36.0 # Required for Mixtral.
|
2023-12-12 09:48:11 +08:00
|
|
|
xformers >= 0.0.23 # Required for CUDA 12.1.
|
2023-05-08 07:30:43 +08:00
|
|
|
fastapi
|
2023-09-28 08:41:36 +08:00
|
|
|
uvicorn[standard]
|
2023-10-22 02:18:58 +08:00
|
|
|
pydantic == 1.10.13 # Required for OpenAI server.
|
2023-12-03 08:37:44 +08:00
|
|
|
aioprometheus[starlette]
|
2023-12-17 13:12:08 +08:00
|
|
|
cupy-cuda12x # Required for CUDA graphs. CUDA 11.8 users should install cupy-cuda11x instead. # FIXME: Fix this in setup.py.
|
