Update modeling_hf_nomic_bert.py

modeling_hf_nomic_bert.py

@@ -1694,6 +1694,7 @@ class NomicBertModel(NomicBertPreTrainedModel):
         return_dict=None,
         matryoshka_dim=None,
         inputs_embeds=None,
+        head_mask=None,
     ):
         if input_ids is not None and inputs_embeds is not None:
             raise ValueError("You cannot specify both input_ids and inputs_embeds at the same time")
@@ -1867,7 +1868,7 @@ class NomicBertForMultipleChoice(NomicBertPreTrainedModel):
     def __init__(self, config):
         super().__init__(config)
 
-        self.new = NomicBertModel(config, add_pooling_layer=True)
+        self.bert = NomicBertModel(config, add_pooling_layer=True)
         classifier_dropout = (
             config.classifier_dropout if config.classifier_dropout is not None else config.hidden_dropout_prob
         )
@@ -1910,7 +1911,7 @@ class NomicBertForMultipleChoice(NomicBertPreTrainedModel):
             else None
         )
 
-        outputs = self.new(
+        outputs = self.bert(
             input_ids,
             attention_mask=attention_mask,
             token_type_ids=token_type_ids,