__init__.py 673 B

1234567891011121314151617181920212223
  1. from .base import (BatchedTensorInputs, 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. "MultiModalDataDict",
  15. "MultiModalInputs",
  16. "MultiModalPlugin",
  17. "MultiModalDataBuiltins",
  18. "NestedTensors",
  19. "MULTIMODAL_REGISTRY",
  20. "MultiModalRegistry",
  21. ]