__init__.py 711 B

123456789101112131415161718192021222324
  1. from .base import (BatchedTensorInputs, BatchedTensors, MultiModalDataBuiltins,
  2. MultiModalDataDict, MultiModalInputs, MultiModalPlugin,
  3. NestedTensors)
  4. from .registry import MultiModalRegistry
  5. MULTIMODAL_REGISTRY = MultiModalRegistry()
  6. """
  7. The global :class:`~MultiModalRegistry` is used by model runners to
  8. dispatch data processing according to its modality and the target model.
  9. See also:
  10. :ref:`input_processing_pipeline`
  11. """
  12. __all__ = [
  13. "BatchedTensorInputs",
  14. "BatchedTensors",
  15. "MultiModalDataDict",
  16. "MultiModalInputs",
  17. "MultiModalPlugin",
  18. "MultiModalDataBuiltins",
  19. "NestedTensors",
  20. "MULTIMODAL_REGISTRY",
  21. "MultiModalRegistry",
  22. ]