click<9.0.0,>=8.0.4
datasets<2.11,>=2.2
filelock
jsonlines
loguru
pydantic
pytorch-lightning<1.10.0,>=1.8.0
rich
sacrebleu
sacremoses
sentencepiece
tensorboardx
torchmetrics<1.0,>=0.9
tokenizers<0.14,>=0.10
tomli
torch<1.14.0,>=1.12.0
transformers!=4.23.0,!=4.23.1,<5.0.0,>=4.19.0

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

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