[Bugfix] Fix unsupported FA version check for Turing GPU (vllm-projec…

…t#12828) Signed-off-by: Felix Marty <[email protected]>
fxmarty-amd · Feb 7, 2025 · 4588bac · 4588bac
1 parent c279d6f
commit 4588bac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vllm/attention/backends/utils.py b/vllm/attention/backends/utils.py
@@ -612,5 +612,5 @@ def flash_attn_version():
         return fa_version
 
     VLLM_FLASH_ATTN_VERSION = flash_attn_version()
-except ImportError:
+except (ImportError, AssertionError):
     VLLM_FLASH_ATTN_VERSION = None