__init__.py 554 B

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