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

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,