[Bugfix] Fix unsupported FA version check for Turing GPU (#12828)

This commit is contained in:
Isotr0py 2025-02-07 01:18:22 +08:00 committed by GitHub
parent afe74f7a96
commit 8108ac841d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -612,5 +612,5 @@ try:
return fa_version
VLLM_FLASH_ATTN_VERSION = flash_attn_version()
except ImportError:
except (ImportError, AssertionError):
VLLM_FLASH_ATTN_VERSION = None