__init__.py 1.11 KB
Newer Older
1
# SPDX-License-Identifier: Apache-2.0
2
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3
from .base import MultiModalPlaceholderMap
4
from .hasher import MultiModalHasher
5
6
from .inputs import (BatchedTensorInputs, ModalityData, MultiModalDataBuiltins,
                     MultiModalDataDict, MultiModalKwargs,
7
                     MultiModalKwargsItems, MultiModalPlaceholderDict,
8
                     MultiModalUUIDDict, NestedTensors)
9
10
11
12
from .registry import MultiModalRegistry

MULTIMODAL_REGISTRY = MultiModalRegistry()
"""
13
14
15
The global [`MultiModalRegistry`][vllm.multimodal.registry.MultiModalRegistry]
is used by model runners to dispatch data processing according to the target
model.
16

17
Info:
18
    [mm_processing](../../../design/mm_processing.html)
19
"""
20
21

__all__ = [
22
    "BatchedTensorInputs",
23
    "ModalityData",
24
    "MultiModalDataBuiltins",
25
    "MultiModalDataDict",
26
    "MultiModalHasher",
27
    "MultiModalKwargs",
28
    "MultiModalKwargsItems",
29
30
    "MultiModalPlaceholderDict",
    "MultiModalPlaceholderMap",
31
    "MultiModalUUIDDict",
32
    "NestedTensors",
33
34
    "MULTIMODAL_REGISTRY",
    "MultiModalRegistry",
35
]