torch>=1.11
transformers>=4.20.1

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