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:
"this method to `embed_input_ids`."
)
model.embed_input_ids = model_get_input_embeddings
return True
logger.warning(
"The model (%s) is missing the `embed_input_ids` method.",
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