Skip to content

Commit d9e625e

Browse files
DarkLight1337lk-chen
authored andcommitted
[Bugfix] Add missing get_language_model to new MLLMs (vllm-project#17300)
Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
1 parent 1f54603 commit d9e625e

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

vllm/model_executor/models/kimi_vl.py

+3
Original file line numberDiff line numberDiff line change
@@ -369,6 +369,9 @@ def _process_image_input(self,
369369
return self.multi_modal_projector(
370370
torch.cat(image_features)).split(lengths)
371371

372+
def get_language_model(self) -> torch.nn.Module:
373+
return self.language_model
374+
372375
def get_multimodal_embeddings(self,
373376
**kwargs: object) -> Optional[NestedTensors]:
374377
# Validate the multimodal input keyword arguments

vllm/model_executor/models/qwen2_5_omni_thinker.py

+3
Original file line numberDiff line numberDiff line change
@@ -809,6 +809,9 @@ def _parse_and_validate_multimodal_inputs(self, **kwargs: object) -> dict:
809809
"audio"] = self._parse_and_validate_audio_input(**kwargs)
810810
return mm_input_by_modality
811811

812+
def get_language_model(self) -> torch.nn.Module:
813+
return self.language_model
814+
812815
def get_multimodal_embeddings(
813816
self, **kwargs: object) -> Optional[MultiModalEmbeddings]:
814817

0 commit comments

Comments
 (0)