Add auto_map to config.json for custom model loading

config.json

@@ -3,6 +3,10 @@
   "architectures": [
     "Imu1ForCausalLM"
   ],
+  "auto_map": {
+    "AutoConfig": "configuration_imu_1.Imu1Config",
+    "AutoModelForCausalLM": "modeling_imu_1.Imu1ForCausalLM"
+  },
   "attn_gating": "per-head",
   "attn_qknorm": true,
   "attn_val_residual": true,