Outlier-150B-V3.2 / configuration_outlier_150b_rexmoe.py
ur-dad-matt's picture
Upload configuration_outlier_150b_rexmoe.py with huggingface_hub
dc24945 verified
Raw
History Blame Contribute Delete
662 Bytes
from transformers import PretrainedConfig
class OutlierReXMoEConfig(PretrainedConfig):
model_type = "outlier_150b_rexmoe"
def __init__(
self,
base_model_name_or_path=None,
moe_layers=None,
n_experts=8,
top_k=2,
psr_scales=None,
group_size=4,
**kwargs,
):
super().__init__(**kwargs)
self.base_model_name_or_path = base_model_name_or_path
self.moe_layers = list(moe_layers or [])
self.n_experts = int(n_experts)
self.top_k = int(top_k)
self.psr_scales = list(psr_scales or [0.7, 0.9, 1.1, 1.3])
self.group_size = int(group_size)