Training checkpoint at step 22000

config.json

@@ -27,5 +27,9 @@
   "transformers_version": "4.57.1",
   "use_denom": true,
   "use_sink": true,
-  "wrap_pos": false
-}
+  "wrap_pos": false,
+  "auto_map": {
+    "AutoConfig": "modeling_armt.ARMTConfig",
+    "AutoModelForCausalLM": "modeling_armt.InnerLoopARMTForCausalLM"
+  }
+}