torch==2.6.0
git+https://github.com/NripeshN/triton.git@main#subdirectory=python
transformers==4.46.3
safetensors==0.4.5