triton==2.1.0
accelerate
transformers
wheel
setuptools