add regex of only tune lm and mm_proj

Former-commit-id: 38d540b3e69bceabafafab524fcfc78aeb05612d
This commit is contained in:
BUAADreamer
2024-05-27 18:59:00 +08:00
parent 11fcd055ec
commit 932f0d5c20
6 changed files with 151 additions and 6 deletions

View File

@@ -10,6 +10,7 @@ from ..extras.logging import get_logger
from .utils.misc import find_all_linear_modules, find_expanded_modules
from .utils.quantization import QuantizationMethod
from .utils.unsloth import get_unsloth_peft_model, load_unsloth_peft_model
from .utils.visual import filter_vision_tower_linear
if TYPE_CHECKING:
@@ -58,6 +59,9 @@ def init_adapter(
if model_args.visual_inputs and hasattr(model, "vision_tower"): # freeze vision model
model.vision_tower.requires_grad_(False)
if model_args.visual_inputs and hasattr(model, "language_model") and model_args.tune_mm_proj: # freeze language model if only tune mm_proj
model.language_model.requires_grad_(False)
if finetuning_args.finetuning_type == "freeze" and is_trainable:
logger.info("Fine-tuning method: Freeze")
num_layers = (
@@ -180,6 +184,9 @@ def init_adapter(
if finetuning_args.use_llama_pro:
target_modules = find_expanded_modules(model, target_modules, finetuning_args.num_layer_trainable)
if model_args.visual_inputs:
target_modules = filter_vision_tower_linear(target_modules)
if (
finetuning_args.use_dora
and getattr(model, "quantization_method", None) is not None