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
|