Update modeling_codet5p.py

modeling_codet5p.py

@@ -758,7 +758,7 @@ def shift_tokens_right(input_ids: torch.Tensor, pad_token_id: int, decoder_start
 # Adapted from transformers.models.encoder_decoder.modeling_encoder_decoder.EncoderDecoderModel
 class CodeT5pEncoderDecoderModel(PreTrainedModel):
     config_class = CodeT5pConfig
-
+    _no_split_modules = ["CodeT5pBlock"]
     def __init__(
             self,
             config: Optional[PretrainedConfig] = None,