rocm skip if flash-attn is installed

This commit is contained in:
Seunghoon Lee
2024-09-26 13:39:47 +09:00
parent 96922d70a7
commit 7b3f4d4020
+2 -1
View File
@@ -563,7 +563,8 @@ def install_rocm_zluda():
if device is not None:
if 'Flash attention' in opts.get('sdp_options'):
install(rocm.get_flash_attention_command(device))
if not installed('flash-attn'):
install(rocm.get_flash_attention_command(device), reinstall=True)
elif not args.experimental:
uninstall('flash-attn')