__init__.py 990 Bytes
Newer Older
1
# SPDX-License-Identifier: Apache-2.0
2
from .base import MultiModalPlaceholderMap
3
from .hasher import MultiModalHashDict, MultiModalHasher
4
5
6
from .inputs import (BatchedTensorInputs, ModalityData, MultiModalDataBuiltins,
                     MultiModalDataDict, MultiModalKwargs,
                     MultiModalPlaceholderDict, NestedTensors)
7
8
9
10
from .registry import MultiModalRegistry

MULTIMODAL_REGISTRY = MultiModalRegistry()
"""
11
12
13
The global [`MultiModalRegistry`][vllm.multimodal.registry.MultiModalRegistry]
is used by model runners to dispatch data processing according to the target
model.
14

15
Info:
16
    [mm_processing](../../../design/mm_processing.html)
17
"""
18
19

__all__ = [
20
    "BatchedTensorInputs",
21
    "ModalityData",
22
    "MultiModalDataBuiltins",
23
    "MultiModalDataDict",
24
25
    "MultiModalHashDict",
    "MultiModalHasher",
26
    "MultiModalKwargs",
27
28
    "MultiModalPlaceholderDict",
    "MultiModalPlaceholderMap",
29
    "NestedTensors",
30
31
    "MULTIMODAL_REGISTRY",
    "MultiModalRegistry",
32
]