torch>=2.5.0
datasets<=3.1.0
transformers>=4.47.0
omegaconf>=2.3.0
accelerate>=0.26.0
pytest
torchao>=0.8.0
