torch
transformers>=4.39.3
datasets
tqdm
numpy
