[Bugfix][v1] Fix step pooler implementation and step pooling usage in v1 (#19956)

Signed-off-by: Isotr0py <2037008807@qq.com>
2025-06-24 02:38:06 +08:00
parent 68aaeb3749
commit 61f4fc5dc6
14 changed files with 164 additions and 40 deletions
--- a/tests/models/language/pooling/embed_utils.py
+++ b/tests/models/language/pooling/embed_utils.py
@@ -55,7 +55,7 @@ def correctness_test_embed_models(hf_runner,
                     task="embed",
                     max_model_len=None,
                     **vllm_extra_kwargs) as vllm_model:
-        vllm_outputs = vllm_model.encode(example_prompts)
+        vllm_outputs = vllm_model.embed(example_prompts)

    with hf_runner(
            model_info.name,