torch
numpy
tqdm
transformers==4.33.1
