torch>=1.7.1
transformers
tqdm
matplotlib
