main
py 12 lines 312 Bytes
Raw
1 # SPDX-License-Identifier: Apache-2.0
2 # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3
4 from ..base.io_processor import PoolingIOProcessor
5
6
7 class ClassifyIOProcessor(PoolingIOProcessor):
8 name = "classify"
9
10
11 class TokenClassifyIOProcessor(PoolingIOProcessor):
12 name = "token_classify"