torch>=1.11
transformers>=4.20.1

[dev]
pytest==6.2.5
pytest-forked
pytest-asyncio==0.16.0
accelerate==0.15.0
black==22.3.0
isort==5.10.1
psutil
peft>=0.3.0
einops==0.6.1
