Hugging Face's logo

PEFT documentation

PEFT

You are viewing v0.6.1 version. A newer version v0.14.0 is available.
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

PEFT

🤗 PEFT, or Parameter-Efficient Fine-Tuning (PEFT), is a library for efficiently adapting pre-trained language models (PLMs) to various downstream applications without fine-tuning all the model’s parameters. PEFT methods only fine-tune a small number of (extra) model parameters, significantly decreasing computational and storage costs because fine-tuning large-scale PLMs is prohibitively costly. Recent state-of-the-art PEFT techniques achieve performance comparable to that of full fine-tuning.

PEFT is seamlessly integrated with 🤗 Accelerate for large-scale models leveraging DeepSpeed and Big Model Inference.

Get started

Start here if you're new to 🤗 PEFT to get an overview of the library's main features, and how to train a model with a PEFT method.
How-to guides

Practical guides demonstrating how to apply various PEFT methods across different types of tasks like image classification, causal language modeling, automatic speech recognition, and more. Learn how to use 🤗 PEFT with the DeepSpeed and Fully Sharded Data Parallel scripts.
Conceptual guides

Get a better theoretical understanding of how LoRA and various soft prompting methods help reduce the number of trainable parameters to make training more efficient.
Reference

Technical descriptions of how 🤗 PEFT classes and methods work.

Supported methods

  1. LoRA: LORA: LOW-RANK ADAPTATION OF LARGE LANGUAGE MODELS
  2. Prefix Tuning: Prefix-Tuning: Optimizing Continuous Prompts for Generation, P-Tuning v2: Prompt Tuning Can Be Comparable to Fine-tuning Universally Across Scales and Tasks
  3. P-Tuning: GPT Understands, Too
  4. Prompt Tuning: The Power of Scale for Parameter-Efficient Prompt Tuning
  5. AdaLoRA: Adaptive Budget Allocation for Parameter-Efficient Fine-Tuning
  6. LLaMA-Adapter: Efficient Fine-tuning of Language Models with Zero-init Attention
  7. IA3: Infused Adapter by Inhibiting and Amplifying Inner Activations

Supported models

The tables provided below list the PEFT methods and models supported for each task. To apply a particular PEFT method for a task, please refer to the corresponding Task guides.

Causal Language Modeling

Model LoRA Prefix Tuning P-Tuning Prompt Tuning IA3
GPT-2 ✅ ✅ ✅ ✅ ✅
Bloom ✅ ✅ ✅ ✅ ✅
OPT ✅ ✅ ✅ ✅ ✅
GPT-Neo ✅ ✅ ✅ ✅ ✅
GPT-J ✅ ✅ ✅ ✅ ✅
GPT-NeoX-20B ✅ ✅ ✅ ✅ ✅
LLaMA ✅ ✅ ✅ ✅ ✅
ChatGLM ✅ ✅ ✅ ✅ ✅

Conditional Generation

Model LoRA Prefix Tuning P-Tuning Prompt Tuning IA3
T5 ✅ ✅ ✅ ✅ ✅
BART ✅ ✅ ✅ ✅ ✅

Sequence Classification

Model LoRA Prefix Tuning P-Tuning Prompt Tuning IA3
BERT ✅ ✅ ✅ ✅ ✅
RoBERTa ✅ ✅ ✅ ✅ ✅
GPT-2 ✅ ✅ ✅ ✅
Bloom ✅ ✅ ✅ ✅
OPT ✅ ✅ ✅ ✅
GPT-Neo ✅ ✅ ✅ ✅
GPT-J ✅ ✅ ✅ ✅
Deberta ✅ ✅ ✅
Deberta-v2 ✅ ✅ ✅

Token Classification

Model LoRA Prefix Tuning P-Tuning Prompt Tuning IA3
BERT ✅ ✅
RoBERTa ✅ ✅
GPT-2 ✅ ✅
Bloom ✅ ✅
OPT ✅ ✅
GPT-Neo ✅ ✅
GPT-J ✅ ✅
Deberta ✅
Deberta-v2 ✅

Text-to-Image Generation

Model LoRA Prefix Tuning P-Tuning Prompt Tuning IA3
Stable Diffusion ✅

Image Classification

Model LoRA Prefix Tuning P-Tuning Prompt Tuning IA3
ViT ✅
Swin ✅

Image to text (Multi-modal models)

We have tested LoRA for ViT and Swin for fine-tuning on image classification. However, it should be possible to use LoRA for any ViT-based model from 🤗 Transformers. Check out the Image classification task guide to learn more. If you run into problems, please open an issue.

Model LoRA Prefix Tuning P-Tuning Prompt Tuning IA3
Blip-2 ✅

Semantic Segmentation

As with image-to-text models, you should be able to apply LoRA to any of the segmentation models. It’s worth noting that we haven’t tested this with every architecture yet. Therefore, if you come across any issues, kindly create an issue report.

Model LoRA Prefix Tuning P-Tuning Prompt Tuning IA3
SegFormer ✅

Quicktour→