__init__.py 646 Bytes
Newer Older
1
from .base import (BatchedTensors, MultiModalDataBuiltins, MultiModalDataDict,
2
                   MultiModalInputs, MultiModalPlugin, NestedTensors)
3
4
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:
    :ref:`input_processing_pipeline`
"""
13
14

__all__ = [
15
    "BatchedTensors",
16
    "MultiModalDataBuiltins",
17
18
    "MultiModalDataDict",
    "MultiModalInputs",
19
    "MultiModalPlugin",
20
    "NestedTensors",
21
22
    "MULTIMODAL_REGISTRY",
    "MultiModalRegistry",
23
]