diff --git a/tensorrt_llm/_torch/speculative/interface.py b/tensorrt_llm/_torch/speculative/interface.py index 3006ccdb4ef..46fe18e0584 100644 --- a/tensorrt_llm/_torch/speculative/interface.py +++ b/tensorrt_llm/_torch/speculative/interface.py @@ -77,7 +77,8 @@ def has_spec_decoder(self): return self.is_mtp() or self.is_eagle3() or self.is_eagle3_one_model() def has_spec_drafter(self): - return self.is_ngram() or self.is_user_provided() + return self.is_eagle3() or self.is_draft_target() or self.is_ngram( + ) or self.is_user_provided() def extend_ctx(self, attention_backend: Type[AttentionBackend]): """