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:
parent
2e1be30ac4
commit
9c7416d834
|
@ -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)
|
||||||
|
|
Loading…
Reference in New Issue