Spaces:
Paused
Paused
Xin Lai
commited on
Commit
·
95dfca2
1
Parent(s):
0077760
Fix initialize_lisa_module during validation in train_ds.py
Browse filesFormer-commit-id: 7232bb8fa764f6d3a9b7928c975e26a67fde545c
- 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 |
-
|
|
|
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
|