[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/test_jina.py
+++ b/tests/models/language/pooling/test_jina.py
@@ -98,11 +98,11 @@ def test_matryoshka(

        if dimensions not in matryoshka_dimensions:
            with pytest.raises(ValueError):
-                vllm_model.encode(
+                vllm_model.embed(
                    example_prompts,
                    pooling_params=PoolingParams(dimensions=dimensions))
        else:
-            vllm_outputs = vllm_model.encode(
+            vllm_outputs = vllm_model.embed(
                example_prompts,
                pooling_params=PoolingParams(dimensions=dimensions))