Upload model files

config.json

@@ -22,9 +22,5 @@
   "tie_word_embeddings": false,
   "transformers_version": "4.57.3",
   "use_cache": true,
-  "vocab_size": 851586,
-  "auto_map": {
-                    "AutoConfig": "modeling_llama_albert.LlamaAlbertConfig",
-                    "AutoModelForCausalLM": "modeling_llama_albert.LlamaAlbertForCausalLM"
-                }
+  "vocab_size": 851586
 }

modeling_llama_albert.py

@@ -13,7 +13,9 @@ class LlamaAlbertConfig(LlamaConfig):
         "AutoConfig": "modeling_llama_albert.LlamaAlbertConfig",
         "AutoModelForCausalLM": "modeling_llama_albert.LlamaAlbertForCausalLM"
     }
+    _auto_class: str = "modeling_llama_albert.LlamaAlbertForCausalLM"
     def __init__(self, embedding_dim=128, **kwargs):
+        print(kwargs)
         super().__init__(**kwargs)
         self.embedding_dim = embedding_dim