Update config.json

config.json

@@ -13,7 +13,7 @@
   "initializer_range": 0.02,
   "intermediate_size": 14336,
   "max_position_embeddings": 32768,
-  "model_type": "llama",  // Changed from "mistral" to "llama"
+  "model_type": "llama",  // Ensure it's "llama"
   "num_attention_heads": 32,
   "num_hidden_layers": 32,
   "num_key_value_heads": 8,
@@ -24,5 +24,9 @@
   "torch_dtype": "float16",
   "transformers_version": "4.47.0",
   "use_cache": true,
-  "vocab_size": 32000
+  "vocab_size": 32000,
+  "auto_map": {
+    "AutoModelForCausalLM": "transformers.models.llama.modeling_llama.LlamaForCausalLM",
+    "AutoTokenizer": "transformers.models.llama.tokenization_llama.LlamaTokenizer"
+  }
 }