pytorch-lightning>=1.2.5
torch>=1.7
PyYAML>=5.1
Pillow>=7.2
torchmetrics>=0.2.0
torchvision>=0.8
transformers>=4.0
pytorch-tabnet==3.1
datasets<1.3,>=1.2
pandas>=1.1
scikit-learn>=0.24
numpy
tqdm
rouge-score>=0.0.4
sentencepiece>=0.1.95
lightning-bolts>=0.3.3
filelock

[:python_version >= "3.7"]
pycocotools>=2.0.2
