zxdu20 richbrain commited on
Commit
9c7416d
1 Parent(s): 2e1be30

fix GLM6BBlock name typo (#20)

Browse files

- fix GLM6BBlock name typo (2a180534c5f2b4860598a9caef798b55a77cfb72)


Co-authored-by: rich brain <richbrain@users.noreply.huggingface.co>

Files changed (1) hide show
  1. 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 = ["GLM6BBlock"]
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)