Update config.json

config.json

@@ -23,5 +23,5 @@
     "v_proj"
   ],
   "task_type": "CAUSAL_LM",
-  "model_type": "TinyLlama"
+  "model_type": "Llama"
 }