ur-dad-matt commited on
Commit
dc24945
·
verified ·
1 Parent(s): 284f0b7

Upload configuration_outlier_150b_rexmoe.py with huggingface_hub

Browse files
configuration_outlier_150b_rexmoe.py ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import PretrainedConfig
2
+
3
+
4
+ class OutlierReXMoEConfig(PretrainedConfig):
5
+ model_type = "outlier_150b_rexmoe"
6
+
7
+ def __init__(
8
+ self,
9
+ base_model_name_or_path=None,
10
+ moe_layers=None,
11
+ n_experts=8,
12
+ top_k=2,
13
+ psr_scales=None,
14
+ group_size=4,
15
+ **kwargs,
16
+ ):
17
+ super().__init__(**kwargs)
18
+ self.base_model_name_or_path = base_model_name_or_path
19
+ self.moe_layers = list(moe_layers or [])
20
+ self.n_experts = int(n_experts)
21
+ self.top_k = int(top_k)
22
+ self.psr_scales = list(psr_scales or [0.7, 0.9, 1.1, 1.3])
23
+ self.group_size = int(group_size)