richbrain commited on
Commit
2a18053
1 Parent(s): 2e1be30

fix GLM6BBlock name typo

Browse files
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)