Update modeling_mistral.py

modeling_mistral.py

@@ -20,9 +20,11 @@ class MistralDenseFormerModel(MistralPreTrainedModel):
     Transformer decoder consisting of *config.num_hidden_layers* layers. Each layer is a [`MistralDecoderLayer`]
 
     Args:
-        config: MistralConfig
+        config: MistralDenseFormerConfig
     """
 
+    config_class = MistralDenseFormerConfig
+
     def __init__(self, config: MistralDenseFormerConfig):
         super().__init__(config)
         self.padding_idx = config.pad_token_id