NameError: name 'FlashMHA' is not defined Killing subprocess 131988
报错NameError: name 'FlashMHA' is not defined Killing subprocess 131988 试图pip install flash_attn报错
您好,请问您是pip install flash_attn失败了吗,需要成功pip install flash_atten后才能通过--use-flash-attention启用FlashAttention。
另外,建议您注意下安装的flash-attn的版本,目前需要在2.0以下,如pip install flash-attn==0.2.8
Running setup.py clean for flash-attn Failed to build flash-attn Installing collected packages: flash-attn Running setup.py install for flash-attn ... error
您好,请问可以提供下完整报错吗?
完整报错太长了,截图截不全
您好,请问您的命令是pip install flash-attn==0.2.8吗,建议您升级Pytorch版本到1.12及以上,另外可以参考一下flash-attn官方仓库的installation部分
@DtYXs 你好,
- 我成功的安装了高版本的flash -attn ,发现咱们预训练模型定义的clip 参数名字跟flash -attn 新版的参数名字不太一致
- 我看即使是低版本的flash -attn 也有这个问题