423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-14 23:58:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update loader.py

Browse Source 
Former-commit-id: 2fc12790414677bb82736208fb9547640780af2e
This commit is contained in:
hoshi-hiyouga 2024-05-11 23:58:47 +08:00 committed by GitHub
parent 4efe56fd68
commit db47c53486
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(