Skip to content

GitLab

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

[Deprecation] Remove missed fallback for `embed_input_ids` (#30469) 


Signed-off-by: default avatarDarkLight1337 <tlleungac@connect.ust.hk>
parent b4e8b912
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 7 deletions
vllm/model_executor/models/interfaces_base.py
View file @ 13d63b65
...@@ -49,13 +49,7 @@ class VllmModel(Protocol[T_co]): ...@@ -49,13 +49,7 @@ class VllmModel(Protocol[T_co]):
def embed_input_ids(self, input_ids: torch.Tensor) -> torch.Tensor: def embed_input_ids(self, input_ids: torch.Tensor) -> torch.Tensor:
"""Apply token embeddings to `input_ids`.""" """Apply token embeddings to `input_ids`."""
if hasattr(self, "get_input_embeddings"): ...
logger.warning_once(
"`get_input_embeddings` for vLLM models is deprecated and will be "
"removed in v0.13.0 or v1.0.0, whichever is earlier. Please rename "
"this method to `embed_input_ids`."
)
return self.get_input_embeddings(input_ids)
def forward(self, input_ids: torch.Tensor, positions: torch.Tensor) -> T_co: ... def forward(self, input_ids: torch.Tensor, positions: torch.Tensor) -> T_co: ...
......
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