[Bug] Fix EmbedIOprocessor "classify" <-> "embed" (#37573)

Signed-off-by: yewentao256 <zhyanwentao@126.com>
This commit is contained in:
Wentao Ye
2026-03-19 19:40:10 -04:00
committed by GitHub
parent 2be1a0f74b
commit df3c0291a3

View File

@@ -23,7 +23,7 @@ def init_pooling_io_processors(
if "embed" in supported_tasks:
from vllm.entrypoints.pooling.embed.io_processor import EmbedIOProcessor
processors.append(("classify", EmbedIOProcessor))
processors.append(("embed", EmbedIOProcessor))
return {
task: processor_cls(