pytorch-lightning==1.7.6
torchvision==0.13.1
torch==1.12.1
protobuf==3.19.5
matplotlib==3.5.3
seaborn==0.12.0
transformers==4.21.3