torch==1.11.0 transformers==4.18 numpy