Update modeling_llama3.py

modeling_llama3.py

@@ -331,12 +331,12 @@ class Llama3TextModel(Llama3PreTrainedModel):
         return causal_mask
 
 class Llama3ForCausalLM(Llama3PreTrainedModel, GenerationMixin):
-    config_class = Llama3ConfigConfig
+    config_class = Llama3Config
     base_model_prefix = "model"
     _tied_weights_keys = ["lm_head.weight"]
 
     def __init__(self, config):
-        super().__init__(config.get_text_config())
+        super().__init__(config)
         self.text_config = config.get_text_config()
         self.vocab_size = self.text_config.vocab_size
         self.model = Llama3TextModel._from_config(config, attn_implementation=config._attn_implementation)