torch
flash-attn!=2.0.9,!=2.1.0,<=2.4.2,>=2.0.6
