accelerate==1.6.0
datasets==3.5.0
numpy>=2.2.4
tokenizers==0.21.1
torch>=2.6.0
transformers==4.50.3
wandb>=0.17.2
tqdm