Unverified Commit f441d36c authored by Johnny Yang's avatar Johnny Yang Committed by GitHub
Browse files

Add missing return in _check_vllm_model_embed_input_ids (#29834)


Signed-off-by: default avatarJohnny Yang <johnnyyang@google.com>
parent 22274b21
...@@ -76,6 +76,7 @@ def _check_vllm_model_embed_input_ids(model: type[object] | object) -> bool: ...@@ -76,6 +76,7 @@ def _check_vllm_model_embed_input_ids(model: type[object] | object) -> bool:
"this method to `embed_input_ids`." "this method to `embed_input_ids`."
) )
model.embed_input_ids = model_get_input_embeddings model.embed_input_ids = model_get_input_embeddings
return True
logger.warning( logger.warning(
"The model (%s) is missing the `embed_input_ids` method.", "The model (%s) is missing the `embed_input_ids` method.",
model, model,
......
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