diff --git a/README.md b/README.md index 45fbf93f76..686400b061 100644 --- a/README.md +++ b/README.md @@ -215,14 +215,15 @@ An example is provided in `~examples/causal_language_modeling/peft_lora_clm_acce ## Models support matrix ### Causal Language Modeling -| Model | LoRA | Prefix Tuning | P-Tuning | Prompt Tuning | -| --------- | ---- | ---- | ---- | ---- | -| GPT-2 | ✅ | ✅ | ✅ | ✅ | -| Bloom | ✅ | ✅ | ✅ | ✅ | -| OPT | ✅ | ✅ | ✅ | ✅ | -| GPT-Neo | ✅ | ✅ | ✅ | ✅ | -| GPT-J | ✅ | ✅ | ✅ | ✅ | -| GPT-NeoX-20B | ✅ | ✅ | ✅ | ✅ | +| Model | LoRA | Prefix Tuning | P-Tuning | Prompt Tuning | +|--------------| ---- | ---- | ---- | ---- | +| GPT-2 | ✅ | ✅ | ✅ | ✅ | +| Bloom | ✅ | ✅ | ✅ | ✅ | +| OPT | ✅ | ✅ | ✅ | ✅ | +| GPT-Neo | ✅ | ✅ | ✅ | ✅ | +| GPT-J | ✅ | ✅ | ✅ | ✅ | +| GPT-NeoX-20B | ✅ | ✅ | ✅ | ✅ | +| LLaMA | ✅ | ✅ | ✅ | ✅ | ### Conditional Generation | Model | LoRA | Prefix Tuning | P-Tuning | Prompt Tuning | diff --git a/src/peft/mapping.py b/src/peft/mapping.py index 68de0c2318..383b3293d4 100644 --- a/src/peft/mapping.py +++ b/src/peft/mapping.py @@ -55,6 +55,7 @@ "deberta-v2": ["query_proj", "value_proj"], "deberta": ["in_proj"], "layoutlm": ["query", "value"], + "llama": ["q_proj", "v_proj"], }