mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-04-24 08:22:07 +00:00
revert unquantized changes
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
This commit is contained in:
parent
bc4eb25d41
commit
b7bdbbd8c0
@ -85,8 +85,6 @@ class UnquantizedSparseMoELayer(nn.Module):
|
|||||||
use_grouped_topk=self.n_expert_group is not None,
|
use_grouped_topk=self.n_expert_group is not None,
|
||||||
num_expert_group=self.n_expert_group,
|
num_expert_group=self.n_expert_group,
|
||||||
topk_group=self.topk_group,
|
topk_group=self.topk_group,
|
||||||
scoring_func=self.scoring_func,
|
|
||||||
e_score_correction_bias=self.e_score_correction_bias,
|
|
||||||
)
|
)
|
||||||
return fused_moe(
|
return fused_moe(
|
||||||
x,
|
x,
|
||||||
|
Loading…
Reference in New Issue
Block a user