aicloud
/
chatglm-6b
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix GLM6BBlock name typo (#20) 

- fix GLM6BBlock name typo (2a180534c5f2b4860598a9caef798b55a77cfb72)


Co-authored-by: rich brain <richbrain@users.noreply.huggingface.co>
This commit is contained in:
Zhengxiao Du 2023-03-31 07:34:06 +00:00 committed by system
parent 2e1be30ac4
commit 9c7416d834
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modeling_chatglm.py
View File

@ -655,7 +655,7 @@ class ChatGLMPreTrainedModel(PreTrainedModel):
supports_gradient_checkpointing = True supports_gradient_checkpointing = True
config_class = ChatGLMConfig config_class = ChatGLMConfig
base_model_prefix = "transformer" base_model_prefix = "transformer"
_no_split_modules = ["GLM6BBlock"] _no_split_modules = ["GLMBlock"]
def __init__(self, *inputs, **kwargs): def __init__(self, *inputs, **kwargs):
super().__init__(*inputs, **kwargs) super().__init__(*inputs, **kwargs)