__init__.py 917 Bytes
Newer Older
1
# SPDX-License-Identifier: Apache-2.0
2
from .base import MultiModalPlaceholderMap
3
from .hasher import MultiModalHashDict, MultiModalHasher
4
5
6
from .inputs import (BatchedTensorInputs, ModalityData, MultiModalDataBuiltins,
                     MultiModalDataDict, MultiModalKwargs,
                     MultiModalPlaceholderDict, NestedTensors)
7
8
9
10
from .registry import MultiModalRegistry

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

14
Info:
15
    [mm-processing][]
16
"""
17
18

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