enable/disable moe token dropping. (#1492)
* Add a flag to enable/disable token dropping in moe/top-1 gating. * fix syntax and formatting.
Showing
想要评论请 注册 或 登录
* Add a flag to enable/disable token dropping in moe/top-1 gating. * fix syntax and formatting.