__init__.py 850 Bytes
Newer Older
1
from .base import MultiModalPlaceholderMap, MultiModalPlugin
2
3
4
from .inputs import (BatchedTensorInputs, ModalityData, MultiModalDataBuiltins,
                     MultiModalDataDict, MultiModalKwargs,
                     MultiModalPlaceholderDict, NestedTensors)
5
6
7
8
9
10
11
12
from .registry import MultiModalRegistry

MULTIMODAL_REGISTRY = MultiModalRegistry()
"""
The global :class:`~MultiModalRegistry` is used by model runners to
dispatch data processing according to its modality and the target model.

See also:
13
    :ref:`input-processing-pipeline`
14
"""
15
16

__all__ = [
17
    "BatchedTensorInputs",
18
    "ModalityData",
19
    "MultiModalDataBuiltins",
20
    "MultiModalDataDict",
21
    "MultiModalKwargs",
22
23
    "MultiModalPlaceholderDict",
    "MultiModalPlaceholderMap",
24
    "MultiModalPlugin",
25
    "NestedTensors",
26
27
    "MULTIMODAL_REGISTRY",
    "MultiModalRegistry",
28
]