katuni4ka commited on
Commit
91af5d5
·
verified ·
1 Parent(s): 8b6d087

Update configuration_mistral4.py

Browse files
Files changed (1) hide show
  1. configuration_mistral4.py +1 -0
configuration_mistral4.py CHANGED
@@ -99,6 +99,7 @@ class Mistral4Config(PretrainedConfig):
99
  self.qk_rope_head_dim = qk_rope_head_dim
100
  self.v_head_dim = v_head_dim
101
  self.qk_nope_head_dim = qk_nope_head_dim
 
102
  self.n_group = n_group
103
  self.topk_group = topk_group
104
  self.num_experts_per_tok = num_experts_per_tok
 
99
  self.qk_rope_head_dim = qk_rope_head_dim
100
  self.v_head_dim = v_head_dim
101
  self.qk_nope_head_dim = qk_nope_head_dim
102
+ self.qk_head_dim = qk_nope_head_dim + qk_nope_head_dim
103
  self.n_group = n_group
104
  self.topk_group = topk_group
105
  self.num_experts_per_tok = num_experts_per_tok