[fix] Fix import error.

2025-12-11 05:31:06 +08:00
parent babfa17354
commit b9ed77cbbb
1 changed files with 1 additions and 1 deletions
--- a/nanovllm/kvcache/chunked_attention.py
+++ b/nanovllm/kvcache/chunked_attention.py
@@ -499,7 +499,7 @@ class ChunkedPrefillState:
 # Test function
 def _test_chunked_attention():
    """Test chunked attention correctness against full attention."""
-    from flash_attn import flash_attn_func
+    from flash_attn.flash_attn_interface import flash_attn_func

    torch.manual_seed(42)