transformers==4.26.0 torch==1.8.1+cu111 sklearn==1.1.2 torchvision==0.9.1+cu111 tqdm==4.64.1 time collections argparse os sys numpy==1.23.1