diff --git a/config.json b/config.json
index 0040623..43075c3 100644
--- a/config.json
+++ b/config.json
@@ -10,7 +10,7 @@
   "initializer_range": 0.02,
   "intermediate_size": 14336,
   "max_position_embeddings": 32768,
-  "model_type": "mistral",
+  "model_type": "llama",
   "num_attention_heads": 32,
   "num_hidden_layers": 32,
   "num_key_value_heads": 8,