ChatGLM support

Signed-off-by: mymusise <mymusise1@gmail.com>
This commit is contained in:
mymusise
2023-03-16 19:09:12 +08:00
parent 64f63a7df2
commit 54b6ce2c0e
2 changed files with 2 additions and 0 deletions
+1
View File
@@ -226,6 +226,7 @@ An example is provided in `~examples/causal_language_modeling/peft_lora_clm_acce
| GPT-J | ✅ | ✅ | ✅ | ✅ |
| GPT-NeoX-20B | ✅ | ✅ | ✅ | ✅ |
| LLaMA | ✅ | ✅ | ✅ | ✅ |
| ChatGLM | ✅ | ✅ | ✅ | ✅ |
### Conditional Generation
| Model | LoRA | Prefix Tuning | P-Tuning | Prompt Tuning |
+1
View File
@@ -56,6 +56,7 @@ TRANSFORMERS_MODELS_TO_LORA_TARGET_MODULES_MAPPING = {
"deberta": ["in_proj"],
"layoutlm": ["query", "value"],
"llama": ["q_proj", "v_proj"],
"chatglm": ["query_key_value"],
}