423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-22 22:02:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update loader.py

Browse Source 
Former-commit-id: 5bfa8e4667534ca7737c5991f59c24ff0d6dc02f
This commit is contained in:
hoshi-hiyouga 2024-05-11 23:58:47 +08:00 committed by GitHub
parent 28d9d7274d
commit 7e0ffc0a64
1 changed files with 1 additions and 2 deletions
Show Stats 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()
else:
model.train()
if model_args.visual_inputs:
model.vision_tower.requires_grad_(False)
trainable_params, all_param = count_parameters(model)
if is_trainable:
param_stats = "trainable params: {:d} || all params: {:d} || trainable%: {:.4f}".format(