python=3.6
torch=1.10.1 (cpu)
transformers=4.18.0
datasets=2.4.0