triton==2.1.0
accelerate
transformers
wheel
setuptools
pycuda==2023.1