From 9c7416d83483e5917034c4db70ec7d5512ac56ca Mon Sep 17 00:00:00 2001
From: Zhengxiao Du <zxdu20@users.noreply.huggingface.co>
Date: Fri, 31 Mar 2023 07:34:06 +0000
Subject: [PATCH] fix GLM6BBlock name typo (#20)

- fix GLM6BBlock name typo (2a180534c5f2b4860598a9caef798b55a77cfb72)


Co-authored-by: rich brain <richbrain@users.noreply.huggingface.co>
---
 modeling_chatglm.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modeling_chatglm.py b/modeling_chatglm.py
index ab51dbb..7f0f0e1 100644
--- a/modeling_chatglm.py
+++ b/modeling_chatglm.py
@@ -655,7 +655,7 @@ class ChatGLMPreTrainedModel(PreTrainedModel):
     supports_gradient_checkpointing = True
     config_class = ChatGLMConfig
     base_model_prefix = "transformer"
-    _no_split_modules = ["GLM6BBlock"]
+    _no_split_modules = ["GLMBlock"]
 
     def __init__(self, *inputs, **kwargs):
         super().__init__(*inputs, **kwargs)