| """Model-agnostic higher-order pruning for PyTorch linear layers.""" | |
| from .core import PruneConfig, PruneReport, discover_linear_groups, prune | |
| from .hf import causal_lm_loss, load_causal_lm, prepare_texts, prune_hf_model | |
| from .pruner import HigherOrderPruner | |
| __all__ = [ | |
| "HigherOrderPruner", | |
| "PruneConfig", | |
| "PruneReport", | |
| "discover_linear_groups", | |
| "prune", | |
| "prune_hf_model", | |
| "load_causal_lm", | |
| "prepare_texts", | |
| "causal_lm_loss", | |
| ] | |