torch>=2.2.0
torchtext>=0.17.0
tqdm
datasets
transformers
