Skip to content

MuhammadBinUsman03/Auto-Prune

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 

Repository files navigation

🔧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.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages