diff --git a/modeling_chatglm.py b/modeling_chatglm.py index bdd2d93..a0f3b9a 100644 --- a/modeling_chatglm.py +++ b/modeling_chatglm.py @@ -610,7 +610,7 @@ class ChatGLMPreTrainedModel(PreTrainedModel): a simple interface for downloading and loading pretrained models. """ - is_parallelizable = True + is_parallelizable = False supports_gradient_checkpointing = False config_class = ChatGLMConfig base_model_prefix = "transformer"