File tree Expand file tree Collapse file tree 1 file changed +3
-1
lines changed
vllm/model_executor/models Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Original file line number Diff line number Diff line change @@ -435,6 +435,7 @@ def __init__(self,
435435 config ,
436436 quant_config = quant_config ,
437437 is_mono = self .is_mono ,
438+ prefix = "vision_model" ,
438439 )
439440
440441 self .language_model = init_vllm_registered_model (
@@ -471,6 +472,7 @@ def _init_vision_model(
471472 quant_config : Optional [QuantizationConfig ],
472473 * ,
473474 is_mono : bool ,
475+ prefix : str ,
474476 ):
475477 if not is_mono :
476478 vision_feature_layer = config .select_layer
@@ -484,7 +486,7 @@ def _init_vision_model(
484486 config .vision_config ,
485487 quant_config = quant_config ,
486488 num_hidden_layers_override = num_hidden_layers ,
487- prefix = "vision_model" ,
489+ prefix = prefix ,
488490 )
489491 else :
490492 return InternVisionPatchModel (config .vision_config )
You can’t perform that action at this time.
0 commit comments