Update modeling_codesage.py

modeling_codesage.py

@@ -349,7 +349,7 @@ class CodeSageForSequenceClassification(CodeSagePreTrainedModel):
         self.num_labels = config.num_labels
         self.config = config
 
-        self.encoder = CodeSageModel(config)
+        self.transformer = CodeSageModel(config)
         classifier_dropout = (
             config.classifier_dropout if config.classifier_dropout is not None else config.residual_dropout_prob
         )
@@ -374,7 +374,7 @@ class CodeSageForSequenceClassification(CodeSagePreTrainedModel):
         return_dict = return_dict if return_dict is not None else self.config.use_return_dict
         assert attention_mask is not None, "attention_mask is needed to perform max-pooling"
 
-        outputs = self.encoder(
+        outputs = self.transformer(
             input_ids,
             attention_mask=attention_mask,
             position_ids=position_ids,