io_processor.py 312 Bytes
Newer Older
1
2
3
# SPDX-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project

4
from ..base.io_processor import PoolingIOProcessor
5
6
7


class ClassifyIOProcessor(PoolingIOProcessor):
8
9
10
11
12
    name = "classify"


class TokenClassifyIOProcessor(PoolingIOProcessor):
    name = "token_classify"