diff --git a/llm/utils.py b/llm/utils.py index 8bcc52ae33ab..d6e3b6a7ab67 100644 --- a/llm/utils.py +++ b/llm/utils.py @@ -125,9 +125,10 @@ def get_lora_target_modules(model): ".*v_proj.*", ".*k_proj.*", ".*o_proj.*", - ".*gate_proj.*", + ".*qkv_proj.*" ".*gate_proj.*", ".*down_proj.*", ".*up_proj.*", + ".*gate_up_fused_proj.*", ] elif model.base_model_prefix == "opt": target_modules = [