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