zai-org
/

chatglm-6b

Model card Files Files and versions

fix GLM6BBlock name typo

#20

by richbrain - opened Mar 31, 2023

base: refs/heads/main

←

from: refs/pr/20

Discussion Files changed

Files changed (1) hide show

modeling_chatglm.py +1 -1

modeling_chatglm.py CHANGED Viewed

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

     supports_gradient_checkpointing = True
     config_class = ChatGLMConfig
     base_model_prefix = "transformer"
+    _no_split_modules = ["GLMBlock"]
     def __init__(self, *inputs, **kwargs):
         super().__init__(*inputs, **kwargs)