Skip to content

Commit 42111d9

Browse files
committed
remove
Signed-off-by: jiahanc <173873397+jiahanc@users.noreply.github.com>
1 parent 84dba6c commit 42111d9

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

vllm/v1/spec_decode/eagle.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,6 @@
2222
from vllm.platforms import current_platform
2323
from vllm.utils import is_pin_memory_available
2424
from vllm.v1.attention.backends.flash_attn import FlashAttentionMetadata
25-
from vllm.v1.attention.backends.flashinfer import FlashInferMetadata
2625
from vllm.v1.attention.backends.tree_attn import (TreeAttentionMetadata,
2726
TreeAttentionMetadataBuilder)
2827
from vllm.v1.attention.backends.triton_attn import TritonAttentionMetadata
@@ -133,8 +132,7 @@ def __init__(
133132
self.allowed_attn_types = tuple(rocm_types)
134133
else:
135134
self.allowed_attn_types = (FlashAttentionMetadata,
136-
TreeAttentionMetadata,
137-
FlashInferMetadata)
135+
TreeAttentionMetadata)
138136

139137
# Parse the speculative token tree.
140138
spec_token_tree = self.speculative_config.speculative_token_tree

0 commit comments

Comments
 (0)