Skip to content

Commit 677db9a

Browse files
Merge pull request #53 from tangtiangu/afd_v0.13.0_fix_precision
[Bugfix] fix precision for afd camp2p
2 parents bfda9f5 + 5ba2a59 commit 677db9a

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

vllm/model_executor/models/deepseek_v2.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1468,8 +1468,11 @@ def compute_attn_output(
14681468
hidden_states=hidden_states,
14691469
router_logits=router_logits,
14701470
top_k=self.top_k,
1471-
use_grouped_topk=False,
1472-
renormalize=True,
1471+
use_grouped_topk=True,
1472+
renormalize=getattr(self.config, "norm_topk_prob", True),
1473+
scoring_func=getattr(self.config, "scoring_func", "softmax"),
1474+
num_expert_group=getattr(self.config, "n_group", 1),
1475+
topk_group = getattr(self.config, "topk_group", 1),
14731476
routed_scaling_factor=1.0 if not mix_placement else routed_scaling_factor,
14741477
e_score_correction_bias=self.gate.e_score_correction_bias,
14751478
mix_placement=mix_placement,

0 commit comments

Comments
 (0)