click<9.0.0,>=8.0.4
datasets<2.19,>=2.18
filelock
jsonlines
loguru
pydantic
pytorch-lightning<2.3.0,>=1.8.0
rich
sacrebleu
sacremoses
sentencepiece
tensorboardx
torchmetrics<2.0,>=0.9
tokenizers<0.16,>=0.10
tomli
torch<2.3,>=2.0
transformers!=4.23.0,!=4.23.1,<5.0.0,>=4.19.0

[lint]
mypy
ruff

[tests]
pytest
pytest-console-scripts
