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(