From df3c0291a3d6aceb0e1393ab0bdbd16dec9f2081 Mon Sep 17 00:00:00 2001 From: Wentao Ye <44945378+yewentao256@users.noreply.github.com> Date: Thu, 19 Mar 2026 19:40:10 -0400 Subject: [PATCH] [Bug] Fix EmbedIOprocessor "classify" <-> "embed" (#37573) Signed-off-by: yewentao256 --- vllm/entrypoints/pooling/io_processor_factories.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/entrypoints/pooling/io_processor_factories.py b/vllm/entrypoints/pooling/io_processor_factories.py index 93ae04bb0..f0c0f5490 100644 --- a/vllm/entrypoints/pooling/io_processor_factories.py +++ b/vllm/entrypoints/pooling/io_processor_factories.py @@ -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(