fix GLM6BBlock name typo (#20)
Browse files- fix GLM6BBlock name typo (2a180534c5f2b4860598a9caef798b55a77cfb72)
Co-authored-by: rich brain <richbrain@users.noreply.huggingface.co>
- modeling_chatglm.py +1 -1
modeling_chatglm.py
CHANGED
@@ -655,7 +655,7 @@ class ChatGLMPreTrainedModel(PreTrainedModel):
|
|
655 |
supports_gradient_checkpointing = True
|
656 |
config_class = ChatGLMConfig
|
657 |
base_model_prefix = "transformer"
|
658 |
-
_no_split_modules = ["
|
659 |
|
660 |
def __init__(self, *inputs, **kwargs):
|
661 |
super().__init__(*inputs, **kwargs)
|
|
|
655 |
supports_gradient_checkpointing = True
|
656 |
config_class = ChatGLMConfig
|
657 |
base_model_prefix = "transformer"
|
658 |
+
_no_split_modules = ["GLMBlock"]
|
659 |
|
660 |
def __init__(self, *inputs, **kwargs):
|
661 |
super().__init__(*inputs, **kwargs)
|