__init__.py 959 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
15
16
17
18
19
20
21
22
23
from .data import (
    DataPrompt,
    DecoderOnlyInputs,
    EmbedsInputs,
    EmbedsPrompt,
    EncoderDecoderInputs,
    ExplicitEncoderDecoderPrompt,
    ProcessorInputs,
    PromptType,
    SingletonInputs,
    SingletonPrompt,
    TextPrompt,
    TokenInputs,
    TokensPrompt,
    build_explicit_enc_dec_prompt,
    embeds_inputs,
    to_enc_dec_tuple_list,
    token_inputs,
    zip_enc_dec_prompts,
)
24
25

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