Do you have a public code sample?

#3
by treehugg3 - opened

Hi, I'm interested in trying my hand at pruning MoE models, with the goal of eventually pruning Kimi-K2. Do you have any code samples for how you are deciding which experts to keep and then actually removing experts from the model? I plan to start with some tiny models first.

Sign up or log in to comment