pytorch ~= 1.8.0
transformers ~= 4.11.3