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

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