__init__.py 754 Bytes
Newer Older
1
from .base import (BatchedTensorInputs, MultiModalDataBuiltins,
2
3
                   MultiModalDataDict, MultiModalInputs, MultiModalPlugin,
                   NestedTensors)
4
from .inputs import MultiModalPlaceholderDict
5
6
7
8
9
10
11
12
13
14
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:
    :ref:`input_processing_pipeline`
"""
15
16

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