miceworks commited on
Commit
c85606c
·
verified ·
1 Parent(s): a890846

feat: upload CONFIGURATION_PY weights (configuration_mark_engineer_v2_8b.py)

Browse files
configuration_mark_engineer_v2_8b.py ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """
2
+ Hugging Face Transformers Configuration for Mark-Engineer-V2-8B
3
+ """
4
+
5
+ from transformers.configuration_utils import PretrainedConfig
6
+
7
+
8
+ class MarkEngineerV2_8BConfig(PretrainedConfig):
9
+ model_type = "mark_engineer_v2_8b"
10
+ keys_to_ignore_at_inference = ["past_key_values"]
11
+
12
+ def __init__(
13
+ self,
14
+ vocab_size: int = 49152,
15
+ hidden_size: int = 2560,
16
+ num_hidden_layers: int = 32,
17
+ num_attention_heads: int = 32,
18
+ num_key_value_heads: int = 8,
19
+ q_lora_rank: int = 1024,
20
+ kv_lora_rank: int = 512,
21
+ qk_nope_head_dim: int = 64,
22
+ qk_rope_head_dim: int = 32,
23
+ v_head_dim: int = 80,
24
+ intermediate_size: int = 7680,
25
+ n_dense_layers: int = 2,
26
+ n_shared_experts: int = 1,
27
+ n_routed_experts: int = 24,
28
+ num_experts_per_tok: int = 2,
29
+ moe_intermediate_size: int = 1254,
30
+ router_bias_update_rate: float = 1e-3,
31
+ mtp_depth: int = 1,
32
+ mtp_loss_weight: float = 0.3,
33
+ max_position_embeddings: int = 8192,
34
+ rope_theta: float = 500000.0,
35
+ rms_norm_eps: float = 1e-5,
36
+ tie_word_embeddings: bool = True,
37
+ bos_token_id: int = 1,
38
+ eos_token_id: int = 2,
39
+ pad_token_id: int = 0,
40
+ **kwargs,
41
+ ):
42
+ super().__init__(
43
+ bos_token_id=bos_token_id,
44
+ eos_token_id=eos_token_id,
45
+ pad_token_id=pad_token_id,
46
+ tie_word_embeddings=tie_word_embeddings,
47
+ **kwargs,
48
+ )
49
+ self.vocab_size = vocab_size
50
+ self.hidden_size = hidden_size
51
+ self.num_hidden_layers = num_hidden_layers
52
+ self.num_attention_heads = num_attention_heads
53
+ self.num_key_value_heads = num_key_value_heads
54
+ self.q_lora_rank = q_lora_rank
55
+ self.kv_lora_rank = kv_lora_rank
56
+ self.qk_nope_head_dim = qk_nope_head_dim
57
+ self.qk_rope_head_dim = qk_rope_head_dim
58
+ self.v_head_dim = v_head_dim
59
+ self.intermediate_size = intermediate_size
60
+ self.n_dense_layers = n_dense_layers
61
+ self.n_shared_experts = n_shared_experts
62
+ self.n_routed_experts = n_routed_experts
63
+ self.num_experts_per_tok = num_experts_per_tok
64
+ self.moe_intermediate_size = moe_intermediate_size
65
+ self.router_bias_update_rate = router_bias_update_rate
66
+ self.mtp_depth = mtp_depth
67
+ self.mtp_loss_weight = mtp_loss_weight
68
+ self.max_position_embeddings = max_position_embeddings
69
+ self.rope_theta = rope_theta
70
+ self.rms_norm_eps = rms_norm_eps