__init__.py 1.22 KB
Newer Older
1
from .base import (BatchedTensorInputs, MultiModalDataBuiltins,
2
                   MultiModalDataDict, MultiModalKwargs,
3
4
                   MultiModalPlaceholderDict, MultiModalPlaceholderMap,
                   MultiModalPlugin, NestedTensors)
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
    "MultiModalDataDict",
20
    "MultiModalKwargs",
21
22
    "MultiModalPlaceholderDict",
    "MultiModalPlaceholderMap",
23
    "MultiModalPlugin",
24
    "NestedTensors",
25
26
    "MULTIMODAL_REGISTRY",
    "MultiModalRegistry",
27
]
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42


def __getattr__(name: str):
    import warnings

    if name == "MultiModalInputs":
        msg = ("MultiModalInputs has been renamed to MultiModalKwargs. "
               "The original name will take another meaning in an upcoming "
               "version.")

        warnings.warn(DeprecationWarning(msg), stacklevel=2)

        return MultiModalKwargs

    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")