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,
@@ -33,4 +34,4 @@
   "torch_dtype": "float32",
   "transformers_version": "4.51.3",
   "vocab_size": 16000
-}
+}