Skip to content

GitLab

Unverified Commit df3c0291 authored by Wentao Ye's avatar Wentao Ye Committed by GitHub
Browse files

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


Signed-off-by: default avataryewentao256 <zhyanwentao@126.com>
parent 2be1a0f7
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
vllm/entrypoints/pooling/io_processor_factories.py
View file @ df3c0291
......@@ -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(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment