click<9.0.0,>=8.0.4
datasets
filelock
loguru
pydantic
pytorch-lightning<1.7.0,>=1.5.6
rich
torch<1.12,>1.8
torchmetrics
tokenizers~=0.10
toml
tqdm
transformers<5.0.0,>=4.0.0

[lint]
black
flakeheaven
flake8-bugbear
flake8-pytest-style
mypy

[tests]
pytest
pytest-console-scripts
pytest-lazy-fixture
