Skip to content

Commit b221923

Browse files
DarkLight1337patrickvonplaten
authored andcommitted
[Model] Implement missing get_language_model for Keye-VL (vllm-project#20631)
Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk> Signed-off-by: Patrick von Platen <patrick.v.platen@gmail.com>
1 parent df802ee commit b221923

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

vllm/model_executor/models/keye.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1592,6 +1592,9 @@ def _parse_and_validate_multimodal_inputs(self, **kwargs: object) -> dict:
15921592

15931593
return modalities
15941594

1595+
def get_language_model(self) -> torch.nn.Module:
1596+
return self.language_model
1597+
15951598
def get_multimodal_embeddings(
15961599
self, **kwargs: object) -> Optional[MultiModalEmbeddings]:
15971600

0 commit comments

Comments
 (0)