torch==2.5.1 triton==3.0.0 transformers==4.46.3 safetensors==0.4.5