accelerate==0.34.2
transformers==4.44.0
datasets==2.18.0
torch==2.1.2
tqdm==4.66.1
numpy==1.26.3
wandb==0.18.6