torch>=2.1.0
transformers>=4.40.0

[all]
datasets>=2.19.0
accelerate>=0.29.0
wandb>=0.16.0

[demo]
datasets>=2.19.0
accelerate>=0.29.0

[dev]
pytest
black
ruff

[wandb]
wandb>=0.16.0
