Skip to content

GitLab

Unverified Commit dd382e0f authored by Cyrus Leung's avatar Cyrus Leung Committed by GitHub
Browse files

[Model] Implement missing `get_language_model` for Keye-VL (#20631) 


Signed-off-by: default avatarDarkLight1337 <tlleungac@connect.ust.hk>
parent 849590a2
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
vllm/model_executor/models/keye.py
View file @ dd382e0f
......@@ -1592,6 +1592,9 @@ class KeyeForConditionalGeneration(nn.Module, SupportsMultiModal, SupportsLoRA,
return modalities
def get_language_model(self) -> torch.nn.Module:
return self.language_model
def get_multimodal_embeddings(
self, **kwargs: object) -> Optional[MultiModalEmbeddings]:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment