torch>=1.4.0
transformers>=4.18.0
numpy>=1.18.2
accelerate
datasets

[dev]
parameterized
pytest
pytest-xdist
black
isort
flake8>=3.8.3
peft>=0.2.0

[peft]
peft>=0.2.0

[test]
parameterized
pytest
pytest-xdist
accelerate
peft
