From 88b0f86c12175c98a4bc735f8fe53f67a9c599fb Mon Sep 17 00:00:00 2001 From: intellinjun Date: Thu, 13 Jun 2024 16:00:14 +0800 Subject: [PATCH] update huggingface.py Signed-off-by: intellinjun --- scripts/huggingface.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/scripts/huggingface.py b/scripts/huggingface.py index 7e977dea7..9a0784441 100644 --- a/scripts/huggingface.py +++ b/scripts/huggingface.py @@ -401,6 +401,10 @@ def __init__( if self.model_type == "chatglm" and "chatglm3" in self._config._name_or_path: # due to the same model architecture. self.model_type = "chatglm2" + # For GLM4 + if model_type == "chatglm" and "glm-4" in self._config._name_or_path: + # due to the same model architecture. + model_type = "chatglm2" @property def config(self):