423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-14 19:06:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update loader.py

Browse Source
This commit is contained in:
hoshi-hiyouga
2024-05-11 23:58:47 +08:00
committed by GitHub
parent bb2e6b0ea3
commit 5bfa8e4667
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/llmtuner/model/loader.py
View File

@@ -155,8 +155,7 @@ def load_model(
model.eval() model.eval()
else: else:
model.train() model.train()
if model_args.visual_inputs:
model.vision_tower.requires_grad_(False)
trainable_params, all_param = count_parameters(model) trainable_params, all_param = count_parameters(model)
if is_trainable: if is_trainable:
param_stats = "trainable params: {:d} || all params: {:d} || trainable%: {:.4f}".format( param_stats = "trainable params: {:d} || all params: {:d} || trainable%: {:.4f}".format(