From 7c985331ddde5bda4d3b250050d75d72586663fe Mon Sep 17 00:00:00 2001 From: roikoren755 <26850796+roikoren755@users.noreply.github.com> Date: Thu, 11 Dec 2025 04:03:50 +0200 Subject: [PATCH] Enable TP for Mamba-based models (#14811) Signed-off-by: Roi Koren --- python/sglang/srt/layers/attention/mamba/mamba.py | 1 - 1 file changed, 1 deletion(-) diff --git a/python/sglang/srt/layers/attention/mamba/mamba.py b/python/sglang/srt/layers/attention/mamba/mamba.py index 6eac7f298..b8d49db97 100644 --- a/python/sglang/srt/layers/attention/mamba/mamba.py +++ b/python/sglang/srt/layers/attention/mamba/mamba.py @@ -339,7 +339,6 @@ class MambaMixer2(torch.nn.Module): input_is_parallel=True, quant_config=quant_config, prefix=f"{prefix}.out_proj", - reduce_results=False, ) self.norm = Mixer2RMSNormGated(