torch>=2.4.0
transformers>=4.44.0

[dev]
pytest==6.2.5
pytest-forked
pytest-asyncio==0.16.0
black==22.3.0
isort==5.13.2

[gpu]
ninja
