Update modeling_hf_nomic_bert.py

modeling_hf_nomic_bert.py

@@ -805,12 +805,12 @@ class NomicBertAttention(nn.Module):
         return attn_output
 
 
-class NomicBertBlock(nn.Module):
+class NomicBertBlock(NomicBertPreTrainedModel):
     def __init__(
         self,
         config,
     ):
-        super().__init__()
+        super().__init__(config=config)
         self.prenorm = config.prenorm
         self.fused_dropout_add_ln = config.fused_dropout_add_ln
 
@@ -1218,4 +1218,4 @@ class NomicBertForSequenceClassification(NomicBertPreTrainedModel):
             logits=logits,
             hidden_states=outputs.hidden_states,
             attentions=outputs.attentions,
-        )
+        )