torch>=1.7.0
transformers
pydantic>=2.0
frozendict

[accelerate]
accelerate

[dev]
black==22.12.0
isort==5.8.0
wheel>=0.36.2
flake8>=3.8.3
pytest>=6.0.0
nbconvert>=7.16.3
