torch==2.5.1+cu121
torchaudio==2.5.1+cu121
transformers==4.49.0
safetensors==0.5.3
numpy==2.1.2
accelerate==1.4.0