Update model architecture to Olmo2

config.json

@@ -1,6 +1,6 @@
 {
   "architectures": [
-    "Olmo1124ForCausalLM"
+    "Olmo2ForCausalLM"
   ],
   "attention_bias": false,
   "attention_dropout": 0.0,
@@ -10,7 +10,7 @@
   "initializer_range": 0.02,
   "intermediate_size": 6144,
   "max_position_embeddings": 4096,
-  "model_type": "olmo_1124",
+  "model_type": "olmo2",
   "num_attention_heads": 16,
   "num_hidden_layers": 16,
   "num_key_value_heads": 16,
@@ -23,4 +23,4 @@
   "transformers_version": "4.47.0.dev0",
   "use_cache": true,
   "vocab_size": 100352
-}
+}