torch>=2.4.1
torchvision>=0.19.1
transformers==4.51.2
sentencepiece
shortuuid
accelerate==1.6.0
peft
bitsandbytes
timm==1.0.10
pydantic
markdown2[all]
numpy
scikit-learn
gradio==4.16.0
gradio_client==0.8.1
deepspeed==0.16.7
requests
httpx==0.24.0
uvicorn
fastapi
einops
einops-exts

[build]
build
twine

[train]
deepspeed==0.16.7
ninja
wandb
