Update configuration_bailing_moe.py
Browse files
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
|