Update config

config.json

@@ -17,7 +17,7 @@
   "num_hidden_layers": 32,
   "num_key_value_heads": 8,
   "num_local_experts": 8,
-  "output_router_logits": true,
+  "output_router_logits": false,
   "rms_norm_eps": 1e-05,
   "rope_theta": 1000000.0,
   "router_aux_loss_coef": 0.02,
@@ -25,6 +25,6 @@
   "tie_word_embeddings": false,
   "torch_dtype": "bfloat16",
   "transformers_version": "4.38.0",
-  "use_cache": false,
+  "use_cache": true,
   "vocab_size": 32000
 }