pytorch-lightning==1.2.0rc0
torch>=1.7
PyYAML>=5.1
Pillow>=7.2
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
pytorch-lightning-bolts==0.3.0
filelock

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