From 865dd03f431c7dd72fff0a1e07105e0e7fc655da Mon Sep 17 00:00:00 2001 From: toyer <2042519524@qq.com> Date: Mon, 1 Jul 2024 03:31:50 +0000 Subject: [PATCH] modified the general name of glm model --- convert-hf-to-gguf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py index a56ce6c88..0838bcb3d 100755 --- a/convert-hf-to-gguf.py +++ b/convert-hf-to-gguf.py @@ -3104,7 +3104,7 @@ class ChatGLMModel(Model): special_vocab.add_to_gguf(self.gguf_writer) def set_gguf_parameters(self): - self.gguf_writer.add_name(self.dir_model.name) + self.gguf_writer.add_name(self.hparams.get("_name_or_path").split("/")[1]) # THUDM/glm4-9b-chat or THUDM/chatglm3-6b n_embed = self.hparams.get("hidden_size", self.hparams.get("n_embed")) n_head = self.hparams.get("n_head", self.hparams.get("num_attention_heads")) n_head_kv = self.hparams.get("multi_query_group_num", n_head)