[Bugfix][Spec Decode] Fix GLM-5.3-Flash DFlash2/EAGLE3 aux hidden-state capture with the upstream config (#60763)

Signed-off-by: Chen Cheng <85005591+ischencheng@users.noreply.github.com> Signed-off-by: Isotr0py <Isotr0py@outlook.com> Co-authored-by: Isotr0py <Isotr0py@outlook.com>
2026-10-09 11:35:48 +00:00
parent 7368fadebd
commit d095e61c0b
1 changed files with 1 additions and 1 deletions
@@ -788,7 +788,7 @@ class Glm5NextModel(nn.Module, EagleModelMixin):
            return hidden_states
        assert residual is not None and comb is not None
        completed = self._aux_post_op(hidden_states, residual, post, comb)
-        return hc_contract(completed, self.config.mhc_num_residual_streams)
+        return hc_contract(completed, self.config.hc_mult)

    def forward(
        self,