mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-01-23 20:24:34 +08:00
Add missing return in _check_vllm_model_embed_input_ids (#29834)
Signed-off-by: Johnny Yang <johnnyyang@google.com>
This commit is contained in:
parent
22274b2184
commit
f441d36cee
@ -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,
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user