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

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

See also:
14
    :ref:`mm-processing`
15
"""
16
17

__all__ = [
18
    "BatchedTensorInputs",
19
    "ModalityData",
20
    "MultiModalDataBuiltins",
21
    "MultiModalDataDict",
22
23
    "MultiModalHashDict",
    "MultiModalHasher",
24
    "MultiModalKwargs",
25
    "MultiModalPlaceholderDict",
26
    "MultiModalPlaceholderMap",
27
    "MultiModalPlugin",
28
    "NestedTensors",
29
30
    "MULTIMODAL_REGISTRY",
    "MultiModalRegistry",
zhuwenwen's avatar
zhuwenwen committed
31
]