fix v1 flash attn get_name

Signed-off-by: Isotr0py <[email protected]>
vllm-project · Nov 23, 2024 · f6cd682 · f6cd682
1 parent 2f9fad5
commit f6cd682
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vllm/v1/attention/backends/flash_attn.py b/vllm/v1/attention/backends/flash_attn.py
@@ -19,7 +19,7 @@ def get_supported_head_sizes() -> List[int]:
 
     @staticmethod
     def get_name() -> str:
-        return "flash-attn-vllm-v1"
+        return "FLASH_ATTN_VLLM_V1"
 
     @staticmethod
     def get_impl_cls() -> Type["FlashAttentionImpl"]: