Update config.json

config.json

@@ -19,6 +19,7 @@
     "v_proj",
     "q_proj"
   ],
-  "task_type": "CAUSAL_LM"
+  "task_type": "CAUSAL_LM",
+  "model_type": "mistral"
 
 }