From 5de9a2b84ff3bbbd6e72375dfcca5c2342da7cd5 Mon Sep 17 00:00:00 2001
From: Isotr0py <2037008807@qq.com>
Date: Thu, 6 Feb 2025 21:05:42 +0800
Subject: [PATCH] handle fa version unsupported

Signed-off-by: Isotr0py <2037008807@qq.com>
---
 vllm/attention/backends/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vllm/attention/backends/utils.py b/vllm/attention/backends/utils.py
index 3c5028a66d58..e8a34434122c 100644
--- 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