From 01ccdb91b162cc399a5bf3a8db3fd43d457bb52b Mon Sep 17 00:00:00 2001
From: MARATRIX <yafeng.li@mthreads.com>
Date: Thu, 26 Mar 2026 12:01:40 +0800
Subject: [PATCH] [Fix] Add EPLB rebalance support for Kimi K2.5 (#21004)

Signed-off-by: yafeng.li <yafeng.li@mthreads.com>
---
 python/sglang/srt/models/kimi_k25.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/python/sglang/srt/models/kimi_k25.py b/python/sglang/srt/models/kimi_k25.py
index 490ef9334..c5bfbac1c 100644
--- a/python/sglang/srt/models/kimi_k25.py
+++ b/python/sglang/srt/models/kimi_k25.py
@@ -767,6 +767,10 @@ class KimiK25ForConditionalGeneration(nn.Module):
     def end_layer(self) -> int:
         return self.language_model.end_layer
 
+    @property
+    def routed_experts_weights_of_layer(self):
+        return self.language_model._routed_experts_weights_of_layer.value
+
     def forward(
         self,
         input_ids: torch.Tensor,