__init__.py 979 Bytes
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
11

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