Xin Lai commited on
Commit
95dfca2
·
1 Parent(s): 0077760

Fix initialize_lisa_module during validation in train_ds.py

Browse files

Former-commit-id: 7232bb8fa764f6d3a9b7928c975e26a67fde545c

Files changed (1) hide show
  1. train_ds.py +2 -1
train_ds.py CHANGED
@@ -161,7 +161,8 @@ def main(args):
161
  model.get_model().initialize_vision_modules(model.get_model().config)
162
  vision_tower = model.get_model().get_vision_tower()
163
  vision_tower.to(dtype=torch_dtype, device=args.local_rank)
164
- model.get_model().initialize_lisa_modules(model.get_model().config)
 
165
 
166
  for p in vision_tower.parameters():
167
  p.requires_grad = False
 
161
  model.get_model().initialize_vision_modules(model.get_model().config)
162
  vision_tower = model.get_model().get_vision_tower()
163
  vision_tower.to(dtype=torch_dtype, device=args.local_rank)
164
+ if not args.eval_only:
165
+ model.get_model().initialize_lisa_modules(model.get_model().config)
166
 
167
  for p in vision_tower.parameters():
168
  p.requires_grad = False