aicloud
/
chatglm-6b
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Set is_parallelizable to False

This commit is contained in:
duzx16 2023-03-16 00:30:43 +08:00
parent 8492687842
commit f9f74fda55
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modeling_chatglm.py
View File

@ -610,7 +610,7 @@ class ChatGLMPreTrainedModel(PreTrainedModel):
a simple interface for downloading and loading pretrained models. a simple interface for downloading and loading pretrained models.
""" """
is_parallelizable = True is_parallelizable = False
supports_gradient_checkpointing = False supports_gradient_checkpointing = False
config_class = ChatGLMConfig config_class = ChatGLMConfig
base_model_prefix = "transformer" base_model_prefix = "transformer"