flash_attn
triton==2.1.0
pycuda==2023.1
accelerate
transformers