diff --git a/python/sglang/srt/speculative/eagle_info.py b/python/sglang/srt/speculative/eagle_info.py index ac629c7ee..c2f90c378 100644 --- a/python/sglang/srt/speculative/eagle_info.py +++ b/python/sglang/srt/speculative/eagle_info.py @@ -154,6 +154,8 @@ class EagleVerifyInput(SpecInput, EagleVerifyInputV2Mixin): dtype=torch.int64, device=batch.device, ) + batch.mamba_track_mask = None + batch.mamba_track_seqlens = None def generate_attn_arg_prefill( self, diff --git a/python/sglang/srt/speculative/eagle_info_v2.py b/python/sglang/srt/speculative/eagle_info_v2.py index 0ae6e3906..d96c89da5 100644 --- a/python/sglang/srt/speculative/eagle_info_v2.py +++ b/python/sglang/srt/speculative/eagle_info_v2.py @@ -242,6 +242,8 @@ class EagleVerifyInputV2Mixin: dtype=torch.int64, device=device, ) + batch.mamba_track_mask = None + batch.mamba_track_seqlens = None # Get a forward batch batch.forward_mode = (