torch==1.10.0 transformers==4.11.3 numpy==1.21.2