Update config.json

config.json

@@ -1 +1,17 @@
-{"model_key": "meta-llama/Llama-3.2-1B", "model_type": "llm"}
+{
+  "architectures": [
+    "LlamaForCausalLM"
+  ],
+  "hidden_size": 4096,
+  "embedding_size": 4096,
+  "num_attention_heads": 32,
+  "num_hidden_layers": 24,
+  "vocab_size": 32000,
+  "max_position_embeddings": 2048,
+  "pad_token_id": 0,
+  "layer_norm_eps": 1e-5,
+  "initializer_range": 0.02,
+  "model_type": "llama",
+  "use_cache": true,
+  "rotary_pos_emb": true
+}