Set is_parallelizable to False
This commit is contained in:
parent
8492687842
commit
f9f74fda55
|
@ -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"
|
||||||
|
|
Loading…
Reference in New Issue