This repository contains tools for pruning transformer blocks in PyTorch:
-
ptblop - package containing builders for block-pruned transformer models in PyTorch
-
ptblopgen - package containing algorithms for block pruning transformer models in PyTorch (work in progress)
-
ptblopgen_evalplus - EvalPlus evaluator plugin for
ptblopgen -
ptblopgen_imagenet - ImageNet evaluator plugin for
ptblopgen