flash-attn==2.6.3
transformers>=4.48.3
accelerate>=1.3.0
einops>=0.8.1
torch>=2.1.0
