From f79ed743f38dac120319d0961b8d39dd3c02bb48 Mon Sep 17 00:00:00 2001 From: "haoning.wu" Date: Sun, 21 Jan 2024 01:32:33 +0800 Subject: [PATCH] fix --- q_align/train/train_mem.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/q_align/train/train_mem.py b/q_align/train/train_mem.py index 638221c..a8d5c1a 100644 --- a/q_align/train/train_mem.py +++ b/q_align/train/train_mem.py @@ -823,8 +823,8 @@ def make_inputs_require_grad(module, input, output): if training_args.freeze_vision_model: for p in model.get_model().vision_model.parameters(): p.requires_grad = False - - model.print_trainable_parameters() + if training_args.lora_enable: + model.print_trainable_parameters() model.config.visual_abstractor_lr = training_args.visual_abstractor_lr