https://download.pytorch.org/whl/cu111/torch-1.9.1%2Bcu111-cp38-cp38-linux_x86_64.whl #torch==1.9.1+cu111
transformers==4.11.3
pytorch-lightning==1.4.9
py-rouge==1.1
nltk==3.6.5
numpy==1.21.2
