torch==1.9.0
pytorch-lightning==1.7.7
transformers
aitextgen==0.6.0