Please install flash attention flash_attn

Please install flash attention

flash_atten安装


pip install flash_attn

linux安装成功:版本 flash-attn-2.7.0.post2

github如果失败:

https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.0.post2/flash_attn-2.7.0.post2+cu12torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl

你可能感兴趣的:(python宝典,python)