__init__.py 1.06 KB
Newer Older
1
# SPDX-License-Identifier: Apache-2.0
2
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3

4
from .data import (DataPrompt, DecoderOnlyInputs, EmbedsInputs, EmbedsPrompt,
5
6
7
8
                   EncoderDecoderInputs, ExplicitEncoderDecoderPrompt,
                   ProcessorInputs, PromptType, SingletonInputs,
                   SingletonPrompt, TextPrompt, TokenInputs, TokensPrompt,
                   build_explicit_enc_dec_prompt, embeds_inputs,
9
                   to_enc_dec_tuple_list, token_inputs, zip_enc_dec_prompts)
10
from .registry import InputContext, InputProcessingContext
11
12

__all__ = [
13
    "DataPrompt",
14
15
    "TextPrompt",
    "TokensPrompt",
16
17
    "PromptType",
    "SingletonPrompt",
18
    "ExplicitEncoderDecoderPrompt",
19
    "TokenInputs",
20
    "EmbedsInputs",
21
    "EmbedsPrompt",
22
    "token_inputs",
23
    "embeds_inputs",
24
25
    "DecoderOnlyInputs",
    "EncoderDecoderInputs",
26
27
    "ProcessorInputs",
    "SingletonInputs",
28
29
30
    "build_explicit_enc_dec_prompt",
    "to_enc_dec_tuple_list",
    "zip_enc_dec_prompts",
31
    "InputContext",
32
    "InputProcessingContext",
33
]