torch==2.2.0
transformers==4.37.2
numpy==1.26.3
tqdm==4.66.1
wandb==0.16.2
