How to install flash attention for vllm 0.5.5?

[brando90]

vllm worked with

pip install torch==2.4.0
pip install vllm==0.5.5 

but was wondering what version of flash attetion would work

# pip install torch==2.2.1 vllm==0.5.1 
# fails install
# pip install flash-attn==2.6.3

fails with 0.5.1 btw.