"benchmarks/vscode:/vscode.git/clone" did not exist on "9798b2fb0052092a6420172e41c0c8a307eedfa6"
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
......@@ -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