Salesforce
/

codet5p-2b

text2text-generation

Model card Files Files and versions

yuewang-sf commited on Aug 4, 2023

Commit

48dd1d6

·

1 Parent(s): 0d030d0

Update modeling_codet5p.py

Files changed (1) hide show

modeling_codet5p.py +1 -0

modeling_codet5p.py CHANGED Viewed

@@ -758,6 +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
     def __init__(
             self,

 # Adapted from transformers.models.encoder_decoder.modeling_encoder_decoder.EncoderDecoderModel
 class CodeT5pEncoderDecoderModel(PreTrainedModel):
     config_class = CodeT5pConfig
+    _no_split_modules = ["CodeT5pBlock"]
     def __init__(
             self,