add AutoModel to auto_map (GLUE classifier loads base via AutoModel)

config.json

@@ -4,7 +4,8 @@
   ],
   "auto_map": {
     "AutoConfig": "configuration_babyloop.BabyloopConfig",
-    "AutoModelForCausalLM": "modeling_babyloop.LoopedForCausalLM"
+    "AutoModelForCausalLM": "modeling_babyloop.LoopedForCausalLM",
+    "AutoModel": "modeling_babyloop.LoopedModel"
   },
   "bias": false,
   "bos_token_id": 1,
@@ -39,4 +40,4 @@
   "visual_inject_iters": 0,
   "visual_inject_mode": "prefix_refresh",
   "vocab_size": 16000
-}
+}