zRzRzRzRzRzRzR commited on
Commit
b4734de
·
1 Parent(s): f226310

add moe_router_dtype config

Browse files
Files changed (1) hide show
  1. config.json +1 -0
config.json CHANGED
@@ -190,6 +190,7 @@
190
  "model_type": "glm_moe_dsa",
191
  "moe_intermediate_size": 2048,
192
  "moe_layer_freq": 1,
 
193
  "n_group": 1,
194
  "n_routed_experts": 256,
195
  "n_shared_experts": 1,
 
190
  "model_type": "glm_moe_dsa",
191
  "moe_intermediate_size": 2048,
192
  "moe_layer_freq": 1,
193
+ "moe_router_dtype": "float32",
194
  "n_group": 1,
195
  "n_routed_experts": 256,
196
  "n_shared_experts": 1,