Training in progress, step 20000

config.json

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