winglian
/

mistral-denseformer-7b

Text Generation

mistral_denseformer

Model card Files Files and versions Community

winglian commited on Mar 23

Commit

97bb87f

•

1 Parent(s): 42b7baa

Update modeling_mistral.py

Files changed (1) hide show

modeling_mistral.py +2 -0

modeling_mistral.py CHANGED Viewed

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

 class MistralDenseFormerForCausalLM(MistralForCausalLM):
+    config_class = MistralDenseFormerConfig
     def __init__(self, config: MistralDenseFormerConfig):
         super().__init__(config)
         self.model = MistralDenseFormerModel(config)