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

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