2023-05-07 16:30:43 -07:00
|
|
|
ninja # For faster builds.
|
|
|
|
|
psutil
|
2023-07-19 22:49:31 -07:00
|
|
|
ray >= 2.5.1
|
2023-09-14 17:31:38 -07:00
|
|
|
pandas # Required for Ray data.
|
2023-09-18 22:36:17 -07:00
|
|
|
pyarrow # Required for Ray data.
|
2023-05-07 16:30:43 -07:00
|
|
|
sentencepiece # Required for LLaMA tokenizer.
|
|
|
|
|
numpy
|
2023-11-16 23:28:39 +01:00
|
|
|
einops # Required for phi-1_5
|
2023-11-08 14:17:49 -08:00
|
|
|
torch >= 2.1.0
|
2023-10-13 10:05:26 -07:00
|
|
|
transformers >= 4.34.0 # Required for Mistral.
|
2023-11-08 14:17:49 -08:00
|
|
|
xformers >= 0.0.22.post7 # Required for CUDA 12.1.
|
2023-05-07 16:30:43 -07:00
|
|
|
fastapi
|
2023-09-27 21:41:36 -03:00
|
|
|
uvicorn[standard]
|
2023-10-21 15:18:58 -03:00
|
|
|
pydantic == 1.10.13 # Required for OpenAI server.
|
