torch>=1.7
transformers==4.31.0
datasets
pytrec_eval
deepspeed==0.10.2
ir_datasets
tqdm
numpy
accelerate==0.26.1