fire
mmengine-lite
numpy
safetensors
sentencepiece
tiktoken
torch
transformers==4.38.2

[all]
pybind11
setuptools
fire
mmengine-lite
numpy
safetensors
sentencepiece
tiktoken
torch
transformers==4.38.2
accelerate
datasets
flash-attn
fastapi
gradio==3.50.2
pydantic>2.0.0
shortuuid
uvicorn

[lite]
accelerate
datasets
flash-attn

[serve]
fastapi
gradio==3.50.2
pydantic>2.0.0
shortuuid
uvicorn
