Update configuration_HelpingAI.py
Browse files
configuration_HelpingAI.py
CHANGED
@@ -40,6 +40,7 @@ class HelpingAIConfig(PretrainedConfig):
|
|
40 |
num_local_experts=8,
|
41 |
router_aux_loss_coef=0.02,
|
42 |
output_router_logits=False,
|
|
|
43 |
**kwargs,
|
44 |
):
|
45 |
self.vocab_size = vocab_size
|
@@ -62,4 +63,5 @@ class HelpingAIConfig(PretrainedConfig):
|
|
62 |
self.num_local_experts = num_local_experts
|
63 |
self.router_aux_loss_coef = router_aux_loss_coef
|
64 |
self.output_router_logits = output_router_logits
|
|
|
65 |
super().__init__(**kwargs)
|
|
|
40 |
num_local_experts=8,
|
41 |
router_aux_loss_coef=0.02,
|
42 |
output_router_logits=False,
|
43 |
+
rope_pct=None,
|
44 |
**kwargs,
|
45 |
):
|
46 |
self.vocab_size = vocab_size
|
|
|
63 |
self.num_local_experts = num_local_experts
|
64 |
self.router_aux_loss_coef = router_aux_loss_coef
|
65 |
self.output_router_logits = output_router_logits
|
66 |
+
self.rope_pct = rope_pct
|
67 |
super().__init__(**kwargs)
|