torch>=2.0.0
tqdm==4.66.5
click==8.1.7
safetensors~=0.4.3
accelerate~=1.0.1
pydantic~=2.9.2
immutables==0.20
transformers>=4.45.2
tokenizers>=0.20.1
huggingface_hub
peft
typing-extensions
sentencepiece
protobuf
scipy
datasets

[dev]
black~=24.10.0
isort~=5.13.2
pre-commit~=4.0.1

[evolve]
ray
cma
lm_eval
wandb

[test]
pytest~=8.3.3

[vllm]
vllm==0.3.2
lm_eval[vllm]
