Commit 6b5ea53c authored by zhuwenwen's avatar zhuwenwen
Browse files

update layer.py

parent 9c4ecf15
......@@ -192,7 +192,7 @@ class UnquantizedFusedMoEMethod(FusedMoEMethodBase, CustomOp):
scoring_func=scoring_func,
e_score_correction_bias=e_score_correction_bias,
activation=activation,
apply_router_weight_on_input=apply_router_weight_on_input
apply_router_weight_on_input=apply_router_weight_on_input,
use_nn_moe=use_nn_moe)
def forward_cuda(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment