Skip to content

Latest commit

 

History

History
30 lines (21 loc) · 1.3 KB

README.md

File metadata and controls

30 lines (21 loc) · 1.3 KB

🔧AutoPrune

LinkedInHugging FaceMedium

simplifies the process of pruning LLMs using a convenient Colab notebook.

Open In Colab

image

🔍 Overview

AutoPrune simplifies the process of pruning LLMs through Colab notebook. You need to specify pruning parameters as described in the pruning paper.

Key Features

  • Automated setup and execution using RunPod.
  • Identifies the most ideal block of layers to prune
  • Removes the pruning block and merges the rest of model
  • Pushed to 🤗 Huggingface the pruned model.
  • Doesn't perform the QLoRA-SFT (Healing) of the pruned model as it can be performed normally.