2024-04-02 13:07:30 +08:00
|
|
|
cmake>=3.21
|
|
|
|
|
ninja # For faster builds.
|
|
|
|
|
psutil
|
|
|
|
|
ray >= 2.9
|
|
|
|
|
sentencepiece # Required for LLaMA tokenizer.
|
|
|
|
|
numpy
|
|
|
|
|
transformers >= 4.38.0 # Required for Gemma.
|
|
|
|
|
fastapi
|
|
|
|
|
uvicorn[standard]
|
|
|
|
|
pydantic >= 2.0 # Required for OpenAI server.
|
|
|
|
|
prometheus_client >= 0.18.0
|
2024-04-04 12:49:49 -07:00
|
|
|
torch == 2.2.1+cpu
|
2024-04-02 13:07:30 +08:00
|
|
|
triton >= 2.1.0
|
|
|
|
|
filelock == 3.13.3
|
|
|
|
|
py-cpuinfo
|
