fix

Q-Future · Jan 20, 2024 · f79ed74 · f79ed74
1 parent d2e350f
commit f79ed74
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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