torch>=1.9
importlib_metadata
filelock
numpy

[dev]
black==22.8
isort>=5.5.4
flake8>=3.8.3
transformers>=4.24.0
accelerate>=0.14.0
datasets>=2.7.0

[quality]
black==22.8
isort>=5.5.4
flake8>=3.8.3
