torch==2.5.1
torchaudio==2.5.1
torchvision==0.20.1
transformers==4.46.1
accelerate==1.0.1
vllm==0.6.4.post1
datasets==3.1.0
wandb==0.17.3 