Model pruning callbacks when using custom distillation lightnignmodule

callbacks
1

If I build a LightningModule to perform distillation (student, teacher), how do I apply ModelPruning callback only to the student model?