diff --git a/installer.py b/installer.py index 25f755a73..c96a10854 100644 --- a/installer.py +++ b/installer.py @@ -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')