Update modeling_hf_nomic_bert.py

modeling_hf_nomic_bert.py

@@ -807,12 +807,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