Skip to content

Commit

Permalink
Move prefix to be closer to variable declaration
Browse files Browse the repository at this point in the history
  • Loading branch information
DarkLight1337 committed Oct 23, 2024
1 parent 0224b75 commit ef4c253
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion vllm/model_executor/models/internvl.py
Original file line number Diff line number Diff line change
Expand Up @@ -435,6 +435,7 @@ def __init__(self,
config,
quant_config=quant_config,
is_mono=self.is_mono,
prefix="vision_model",
)

self.language_model = init_vllm_registered_model(
Expand Down Expand Up @@ -471,6 +472,7 @@ def _init_vision_model(
quant_config: Optional[QuantizationConfig],
*,
is_mono: bool,
prefix: str,
):
if not is_mono:
vision_feature_layer = config.select_layer
Expand All @@ -484,7 +486,7 @@ def _init_vision_model(
config.vision_config,
quant_config=quant_config,
num_hidden_layers_override=num_hidden_layers,
prefix="vision_model",
prefix=prefix,
)
else:
return InternVisionPatchModel(config.vision_config)
Expand Down

0 comments on commit ef4c253

Please sign in to comment.