Skip to content

Commit 252fdc4

Browse files
committed
[LLM] fix lora target modules on llama
1 parent fdcabf8 commit 252fdc4

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

llm/utils.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -125,9 +125,10 @@ def get_lora_target_modules(model):
125125
".*v_proj.*",
126126
".*k_proj.*",
127127
".*o_proj.*",
128-
".*gate_proj.*",
128+
".*qkv_proj.*" ".*gate_proj.*",
129129
".*down_proj.*",
130130
".*up_proj.*",
131+
".*gate_up_fused_proj.*",
131132
]
132133
elif model.base_model_prefix == "opt":
133134
target_modules = [

0 commit comments

Comments
 (0)