How to prune a model trained with LayerDrop

🤗Transformers
1

I am training a BART transformers with LayerDrop for encoder and decoder.Now, I want to know how can I do the pruning? I did not find enough information in the documentation.