fix logging msg for block manager (#3701)

2024-03-28 16:29:55 -07:00
parent c0935c96d3
commit 4716a32dd4
3 changed files with 5 additions and 4 deletions
--- a/vllm/attention/selector.py
+++ b/vllm/attention/selector.py
@@ -41,6 +41,8 @@ def _can_use_flash_attn(dtype: torch.dtype) -> bool:
    try:
        import flash_attn  # noqa: F401
    except ImportError:
-        logger.info("flash_attn is not found.")
+        logger.info(
+            "Cannot use FlashAttention because the package is not found. "
+            "Please install it for better performance.")
        return False
    return True