codesage
/

codesage-small

Inference Endpoints

Model card Files Files and versions Community

codesage commited on Feb 20

Commit

d4c8b2d

•

1 Parent(s): 542998f

Update modeling_codesage.py

Files changed (1) hide show

modeling_codesage.py +3 -1

modeling_codesage.py CHANGED Viewed

@@ -351,7 +351,9 @@ class CodeSageForSequenceClassification(CodeSagePreTrainedModel):
         self.transformer = CodeSageModel(config)
         classifier_dropout = (
-            config.classifier_dropout if config.classifier_dropout is not None else config.residual_dropout_prob
         )
         self.dropout = nn.Dropout(classifier_dropout)
         self.classifier = nn.Linear(config.hidden_size, config.num_labels)

         self.transformer = CodeSageModel(config)
         classifier_dropout = (
+            config.classifier_dropout
+            if hasattr(config, 'classifier_dropout') and config.classifier_dropout is not None
+            else config.residual_dropout_prob
         )
         self.dropout = nn.Dropout(classifier_dropout)
         self.classifier = nn.Linear(config.hidden_size, config.num_labels)