Skip to content

Commit ef4c253

Browse files
committed
Move prefix to be closer to variable declaration
1 parent 0224b75 commit ef4c253

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

vllm/model_executor/models/internvl.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff 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)

0 commit comments

Comments
 (0)