zhujiangang commited on
Commit
12db5bf
·
verified ·
1 Parent(s): 714e5fa

Update configuration_bailing_moe.py

Browse files
Files changed (1) hide show
  1. configuration_bailing_moe.py +1 -1
configuration_bailing_moe.py CHANGED
@@ -63,7 +63,7 @@ class BailingMoeConfig(PretrainedConfig):
63
  self.use_sliding_window = use_sliding_window
64
  self.sliding_window = sliding_window
65
  self.max_window_layers = max_window_layers
66
- self.head_dim = head_dim
67
  self.rope_scaling = rope_scaling
68
 
69
  # MoE configs
 
63
  self.use_sliding_window = use_sliding_window
64
  self.sliding_window = sliding_window
65
  self.max_window_layers = max_window_layers
66
+ self.head_dim = head_dim or self.hidden_size // self.num_attention_heads
67
  self.rope_scaling = rope_scaling
68
 
69
  # MoE configs