diff --git a/python/sglang/srt/speculative/eagle_worker_v2.py b/python/sglang/srt/speculative/eagle_worker_v2.py index 80526c3f5..bc2d7f183 100644 --- a/python/sglang/srt/speculative/eagle_worker_v2.py +++ b/python/sglang/srt/speculative/eagle_worker_v2.py @@ -471,6 +471,11 @@ class EagleDraftWorker(EagleDraftWorkerBase): ) graph_supported_backend_types.append(DeepseekSparseAttnBackend) + from sglang.srt.layers.attention.deepseek_v4_backend import ( + DeepseekV4AttnBackend, + ) + + graph_supported_backend_types.append(DeepseekV4AttnBackend) graph_supported_backend = isinstance( self.draft_extend_attn_backend,