From 6ea69efb7f65acfbad5dda47464b0cc8e7f01d0e Mon Sep 17 00:00:00 2001 From: Byron Hsu Date: Fri, 29 May 2026 15:08:58 -0700 Subject: [PATCH] [RL] Forward Kimi K2.5 weight hooks to language model (#26744) Co-authored-by: Byron Hsu <24364830+ByronHsu@users.noreply.github.com> --- python/sglang/srt/models/kimi_k25.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/python/sglang/srt/models/kimi_k25.py b/python/sglang/srt/models/kimi_k25.py index b4a264848..c270b9b7f 100644 --- a/python/sglang/srt/models/kimi_k25.py +++ b/python/sglang/srt/models/kimi_k25.py @@ -804,6 +804,24 @@ class KimiK25ForConditionalGeneration(nn.Module): for _ in stream_language_weights(): pass + def post_load_weights(self): + if self.language_model is not None: + self.language_model.post_load_weights() + + @property + def stacked_params_mapping(self): + return getattr(self.language_model, "stacked_params_mapping", []) + + @property + def expert_params_mapping(self): + return getattr(self.language_model, "expert_params_mapping", []) + + def mutate_weight_preload(self, name): + return self.language_model.mutate_weight_preload(name) + + def custom_scale_remap(self, name): + return self.language_model.custom_scale_remap(name) + @classmethod def get_model_config_for_expert_location(cls, config: KimiK25Config): text_config = config.text_config