torch<=2.9.1,>=1.7.0
transformers<5.0.0
pydantic>=2.0
loguru

[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
transformers<5.0
accelerate
