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-05-08 07:30:43 +08:00
|
|
|
sentencepiece # Required for LLaMA tokenizer.
|
|
|
|
|
numpy
|
|
|
|
|
torch >= 2.0.0
|
2023-07-21 02:38:27 +08:00
|
|
|
transformers >= 4.31.0 # Required for LLaMA-2.
|
2023-08-23 16:44:21 +08:00
|
|
|
xformers >= 0.0.21
|
2023-05-08 07:30:43 +08:00
|
|
|
fastapi
|
|
|
|
|
uvicorn
|
2023-07-12 23:10:55 +08:00
|
|
|
pydantic < 2 # Required for OpenAI server.
|
