Commit 46b099a

authored and

committed

Add MoE pruning support.

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

1 parent 009bd1a commit 46b099aCopy full SHA for 46b099a

2 files changed

+298

-32

lines changed

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

+298

-32

lines changed

Comments

(0)