torch>=2.0.0
tqdm==4.67.1
click==8.2.1
safetensors~=0.5.2
accelerate~=1.6.0
pydantic~=2.10.6
immutables==0.21
transformers>=4.45.2
tokenizers>=0.20.1
huggingface_hub
peft
typing-extensions
sentencepiece
protobuf
scipy
datasets

[dev]
black~=25.1.0
isort~=6.0.1
pre-commit~=4.2.0

[evolve]
ray
cma
lm_eval
wandb

[test]
pytest~=8.4.0

[vllm]
vllm==0.7.2
lm_eval[vllm]
