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

4
5
6
7
8
9
10
11
12
13
14
from .data import (
    DataPrompt,
    DecoderOnlyInputs,
    EmbedsInputs,
    EmbedsPrompt,
    EncoderDecoderInputs,
    ExplicitEncoderDecoderPrompt,
    ProcessorInputs,
    PromptType,
    SingletonInputs,
    SingletonPrompt,
15
    StreamingInput,
16
17
18
19
20
21
22
23
24
    TextPrompt,
    TokenInputs,
    TokensPrompt,
    build_explicit_enc_dec_prompt,
    embeds_inputs,
    to_enc_dec_tuple_list,
    token_inputs,
    zip_enc_dec_prompts,
)
25
26

__all__ = [
27
    "DataPrompt",
28
29
    "TextPrompt",
    "TokensPrompt",
30
31
    "PromptType",
    "SingletonPrompt",
32
    "ExplicitEncoderDecoderPrompt",
33
    "TokenInputs",
34
    "EmbedsInputs",
35
    "EmbedsPrompt",
36
    "token_inputs",
37
    "embeds_inputs",
38
39
    "DecoderOnlyInputs",
    "EncoderDecoderInputs",
40
41
    "ProcessorInputs",
    "SingletonInputs",
42
43
44
    "build_explicit_enc_dec_prompt",
    "to_enc_dec_tuple_list",
    "zip_enc_dec_prompts",
45
    "StreamingInput",
46
]