Johnblick187 commited on
Commit
e3541e6
·
verified ·
1 Parent(s): bcf87fa

Update modeling_smartcoder_moe.py

Browse files
Files changed (1) hide show
  1. modeling_smartcoder_moe.py +1 -1
modeling_smartcoder_moe.py CHANGED
@@ -34,7 +34,7 @@ class SmartCoderMoEConfig(PretrainedConfig):
34
  num_key_value_heads=4,
35
  dense_intermediate_size=8192,
36
  num_experts=32,
37
- expert_intermediate_size=512,
38
  num_experts_per_tok=2,
39
  max_position_embeddings=16384,
40
  rope_theta=10000.0,
 
34
  num_key_value_heads=4,
35
  dense_intermediate_size=8192,
36
  num_experts=32,
37
+ expert_dim=512,
38
  num_experts_per_tok=2,
39
  max_position_embeddings=16384,
40
  rope_theta=10000.0,