Commit 3d246d7

authored and

committed

Add MoE pruning support.

Signed-off-by: Keval Morabia <[email protected]>

1 parent 8cf516e commit 3d246d7Copy full SHA for 3d246d7

2 files changed

+298

-32

lines changed

modelopt/torch
- nas/plugins
  - megatron.py
- prune/plugins
  - mcore_minitron.py

+298

-32

lines changed

Comments

(0)