Add missing return in _check_vllm_model_embed_input_ids (#29834)

Signed-off-by: Johnny Yang <johnnyyang@google.com>
This commit is contained in:
Johnny Yang 2025-12-01 19:22:50 -08:00 committed by GitHub
parent 22274b2184
commit f441d36cee
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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,