minor fix and cleanup

This commit is contained in:
Sigbjørn Skjæret 2026-02-13 11:57:45 +01:00 committed by GitHub
parent 7b23cd9207
commit 1daef5f85f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 2 deletions

View File

@ -8580,7 +8580,7 @@ class Glm4MoeModel(TextModel):
self.tensor_map = gguf.get_tensor_name_map(self.model_arch, self.block_count)
def set_vocab(self):
Glm4MoeModel.set_vocab_glm(self)
return self._set_vocab_glm()
def set_gguf_parameters(self):
super().set_gguf_parameters()
@ -8691,7 +8691,6 @@ class GlmMoeDsaModel(DeepseekV2Model):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
# GLM4_MOE has num_hidden_layers + 1 actual layers (including NextN layer)
self.block_count = self.hparams["num_hidden_layers"] + self.hparams.get("num_nextn_predict_layers", 0)
self.tensor_map = gguf.get_tensor_name_map(self.model_arch, self.block_count)