torch>=2.0
transformers>=4.40
sentencepiece>=0.1.99
numpy>=1.21
pandas>=1.5
scikit-learn>=1.2
pyyaml>=6.0
tqdm>=4.65
datasets>=2.14
pydantic>=2.0
gradio>=4.0
litellm>=1.0
peft>=0.7
torch-geometric>=2.3
scipy>=1.10
protobuf>=3.20
fastapi
uvicorn
httpx

[all]
vllm==0.6.3
torch==2.4.0
openai>=1.0

[router-r1]
vllm==0.6.3
torch==2.4.0
openai>=1.0
