update huggingface.py

Signed-off-by: intellinjun <jun.lin@intel.com>
intel · Jun 13, 2024 · 88b0f86 · 88b0f86
1 parent 987b050
commit 88b0f86
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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):