python == 3.8
torch == 1.8.2
transformers == 4.16.0
datasets == 1.18.3