click<9.0.0,>=8.0.4
datasets<2.20,>=2.18
filelock
jsonlines
lightning<2.3.0,>=2.0.0
loguru
pydantic
pytorch-lightning<2.3.0,>=2.0.0
rich
sacrebleu
sacremoses
sentencepiece
system_info
tensorboardx
torchmetrics<2.0,>=0.9
tokenizers<0.20,>=0.10
tomli
torch<2.4,>=2.0
transformers!=4.23.0,!=4.23.1,<4.41.0,>=4.19.0

[lint]
mypy
ruff

[tests]
pytest
pytest-console-scripts
