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

4
5
6
7
8
from .data import (DecoderOnlyInputs, EmbedsInputs, EmbedsPrompt,
                   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
from .registry import (DummyData, InputContext, InputProcessingContext,
                       InputRegistry)
12
13
14

INPUT_REGISTRY = InputRegistry()
"""
15
16
17
The global [`InputRegistry`][vllm.inputs.registry.InputRegistry] which is used
by [`LLMEngine`][vllm.LLMEngine] to dispatch data processing according to the
target model.
18
19
20
"""

__all__ = [
21
22
    "TextPrompt",
    "TokensPrompt",
23
24
    "PromptType",
    "SingletonPrompt",
25
    "ExplicitEncoderDecoderPrompt",
26
    "TokenInputs",
27
    "EmbedsInputs",
28
    "EmbedsPrompt",
29
    "token_inputs",
30
    "embeds_inputs",
31
32
    "DecoderOnlyInputs",
    "EncoderDecoderInputs",
33
34
    "ProcessorInputs",
    "SingletonInputs",
35
36
37
    "build_explicit_enc_dec_prompt",
    "to_enc_dec_tuple_list",
    "zip_enc_dec_prompts",
38
    "INPUT_REGISTRY",
39
    "DummyData",
40
    "InputContext",
41
    "InputProcessingContext",
42
    "InputRegistry",
43
]