diff --git a/python/sglang/srt/models/deepseek_v2.py b/python/sglang/srt/models/deepseek_v2.py index 87d751465..24b46c0bc 100644 --- a/python/sglang/srt/models/deepseek_v2.py +++ b/python/sglang/srt/models/deepseek_v2.py @@ -326,7 +326,7 @@ class MoEGate(nn.Module): and (self.weight.shape[0] == 256 or self.weight.shape[0] == 384) and _device_sm >= 90 ): - if _device_sm >= 100 and self.weight.shape[0] == 256: + if _device_sm == 100 and self.weight.shape[0] == 256: # router gemm output float32 logits = torch.empty( hidden_states.shape[0],