update

config.json

@@ -1,11 +1,11 @@
 {
   "_name_or_path": "Magma-8B",
   "architectures": [
-    "MagmaForConditionalGeneration"
+    "MagmaForCausalLM"
   ],
   "auto_map": {
     "AutoConfig": "configuration_magma.MagmaConfig",
-    "AutoModelForCausalLM": "modeling_magma.MagmaForForCausalLM"
+    "AutoModelForCausalLM": "modeling_magma.MagmaForCausalLM"
   },
   "attention_bias": false,
   "attention_dropout": 0.0,

modeling_magma.py

@@ -251,7 +251,7 @@ MAGMA_INPUTS_DOCSTRING = r"""
     """The Magma model which consists of a vision backbone and a language model.""",
     MAGMA_START_DOCSTRING,
 )
-class MagmaForForCausalLM(MagmaPreTrainedModel):
+class MagmaForCausalLM(MagmaPreTrainedModel):
     def __init__(self, config: MagmaConfig):
         super().__init__(config)