Update vllm_ascend/attention.py

Co-authored-by: Mengqing Cao <cmq0113@163.com>
cosdt · Jan 13, 2025 · 98b01b8 · 98b01b8
1 parent dc63477
commit 98b01b8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vllm_ascend/attention.py b/vllm_ascend/attention.py
@@ -19,7 +19,7 @@
                                            PagedAttentionMetadata)
 
 if TYPE_CHECKING:
-    from vllm_ascend.worker import ModelInputForNPUBuilder
+    from vllm_ascend.model_runner import ModelInputForNPUBuilder
 
 SHARE_MASK_TRIL_PREFIX_CACHE = None
 SHARE_MASK_TRIL = None