initial upload

config.json

@@ -1,17 +1,20 @@
 {
-  "_name_or_path": "./tinyllama_08b/ckpt_81",
+  "_name_or_path": "MobiLlama",
   "architectures": [
-    "LlamaForCausalLM"
+    "MobiLlamaForCausalLM"
   ],
+  "auto_map": {
+    "AutoModelForCausalLM": "modelling_mobillama.MobiLlamaForCausalLM"
+  },
   "attention_bias": false,
   "attention_dropout": 0.0,
   "bos_token_id": 1,
   "eos_token_id": 2,
   "hidden_act": "silu",
-  "hidden_size": 2560,
+  "hidden_size": 2048,
   "initializer_range": 0.02,
-  "intermediate_size": 10240,
-  "max_position_embeddings": 2560,
+  "intermediate_size": 5632,
+  "max_position_embeddings": 2048,
   "model_type": "llama",
   "num_attention_heads": 32,
   "num_hidden_layers": 22,
@@ -25,4 +28,4 @@
   "transformers_version": "4.36.1",
   "use_cache": true,
   "vocab_size": 32000
-}
+}