Update modeling_mistral.py

modeling_mistral.py

@@ -200,6 +200,8 @@ class MistralDenseFormerModel(MistralPreTrainedModel):
 
 
 class MistralDenseFormerForCausalLM(MistralForCausalLM):
+    config_class = MistralDenseFormerConfig
+
     def __init__(self, config: MistralDenseFormerConfig):
         super().__init__(config)
         self.model = MistralDenseFormerModel(config)