diff --git a/vllm/model_executor/models/interfaces_base.py b/vllm/model_executor/models/interfaces_base.py index 2c99fce8d918c..e8d521ec2e8aa 100644 --- a/vllm/model_executor/models/interfaces_base.py +++ b/vllm/model_executor/models/interfaces_base.py @@ -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,