File tree Expand file tree Collapse file tree 1 file changed +4
-0
lines changed Expand file tree Collapse file tree 1 file changed +4
-0
lines changed Original file line number Diff line number Diff line change @@ -303,6 +303,8 @@ def load_weights(
303
303
# Skip loading extra bias for GPTQ models.
304
304
if name .endswith (".bias" ) and name not in params_dict :
305
305
continue
306
+ if name .startswith ("model.vision_tower" ) and name not in params_dict :
307
+ continue
306
308
param = params_dict [name ]
307
309
weight_loader = param .weight_loader
308
310
weight_loader (param , loaded_weight , shard_id )
@@ -311,6 +313,8 @@ def load_weights(
311
313
# Skip loading extra bias for GPTQ models.
312
314
if name .endswith (".bias" ) and name not in params_dict :
313
315
continue
316
+ if name .startswith ("model.vision_tower" ) and name not in params_dict :
317
+ continue
314
318
param = params_dict [name ]
315
319
weight_loader = getattr (param , "weight_loader" , default_weight_loader )
316
320
weight_loader (param , loaded_weight )
You can’t perform that action at this time.
0 commit comments