[Embedding] Recover embedding function (#2483)

Fix broken embedding function. It's broken by
http://github.com/vllm-project/vllm/pull/23162

- vLLM version: v0.10.1.1
- vLLM main:
efc88cf64a

Signed-off-by: wangxiyuan <wangxiyuan1007@gmail.com>
This commit is contained in:
wangxiyuan
2025-08-27 09:22:01 +08:00
committed by GitHub
parent 6a4ec186e7
commit f22077daa6
3 changed files with 129 additions and 58 deletions

View File

@@ -19,7 +19,6 @@
from collections.abc import Sequence
from typing import Optional
import pytest
from modelscope import snapshot_download # type: ignore[import-untyped]
from tests.e2e.conftest import HfRunner
@@ -50,8 +49,6 @@ def test_dummy():
assert True
@pytest.mark.skip(
reason="TODO: revert me when pooler is adapted with the latest vllm main")
def test_embed_models_correctness(hf_runner, vllm_runner):
queries = ['What is the capital of China?', 'Explain gravity']