triton==2.1.0
accelerate
transformers
flash_attn
pycuda==2023.1