Update config.json

config.json

@@ -25,5 +25,7 @@
   "torch_dtype": "float16",
   "transformers_version": "4.34.0",
   "use_cache": true,
-  "vocab_size": 32000
+  "vocab_size": 32000,
+  "medusa_num_heads": 5,
+  "medusa_num_layers": 1
 }