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

…t#12828)
drikster80 · Feb 6, 2025 · 8108ac8 · 8108ac8
1 parent afe74f7
commit 8108ac8
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