Signed-off-by: wang.yuqi <yuqi.wang@daocloud.io> Signed-off-by: wang.yuqi <noooop@126.com> Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
13 lines
312 B
Python
13 lines
312 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
|
|
from ..base.io_processor import PoolingIOProcessor
|
|
|
|
|
|
class ClassifyIOProcessor(PoolingIOProcessor):
|
|
name = "classify"
|
|
|
|
|
|
class TokenClassifyIOProcessor(PoolingIOProcessor):
|
|
name = "token_classify"
|