From 4e571ce6433b6768950becda40d55cb4f24741ce Mon Sep 17 00:00:00 2001 From: Matthew Bonanni Date: Mon, 9 Mar 2026 14:43:06 -0400 Subject: [PATCH] [MTP][Misc] Clean up dead code (#36507) Signed-off-by: Matthew Bonanni --- vllm/v1/spec_decode/eagle.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/vllm/v1/spec_decode/eagle.py b/vllm/v1/spec_decode/eagle.py index d05895b18..89c9c80ce 100644 --- a/vllm/v1/spec_decode/eagle.py +++ b/vllm/v1/spec_decode/eagle.py @@ -481,10 +481,7 @@ class SpecDecodeBaseProposer: positions = self.mrope_positions[:, token_indices_to_sample] else: positions = self.positions[token_indices_to_sample] - if self.method == "mtp": - hidden_states = self.hidden_states[token_indices_to_sample] - else: - hidden_states = hidden_states[token_indices_to_sample] + hidden_states = hidden_states[token_indices_to_sample] if isinstance(attn_metadata, TreeAttentionMetadata): # Draft using tree attention - requires full logits for top-k