Update modeling_lsg_mbart.py

modeling_lsg_mbart.py

@@ -673,10 +673,6 @@ class LSGMBartPretrainedModel(MBartPreTrainedModel):
     base_model_prefix = "model"
     supports_gradient_checkpointing = True
 
-    def _set_gradient_checkpointing(self, module, value=False):
-        if isinstance(module, (MBartDecoder, MBartEncoder, LSGMBartEncoder)):
-            module.gradient_checkpointing = value
-
 
 class LSGMBartEncoder(LSGMBartPretrainedModel, MBartEncoder):
     """