Evo installation with flash-attn==1.0.5

[ItamarElkayam]

Hey!
After struggling with the "No Module Named 'torch'" issue, I installed flash-attn==1.0.5 as suggested. However, when I try to install evo, it attempts to reinstall a newer version of flash-attn and fails.

I also tried manually installing all requirements but ran into:
TypeError: MHA.init() got an unexpected keyword argument 'num_heads_kv'

How can I proceed with evo's installation after manually installing flash-attn?

Thanks!