bridge.data.collators.visual

Shared visual tensor key groups for model-owned collators.

Module Contents

Data

THW_GRID_VISUAL_KEYS
PASSTHROUGH_VISUAL_KEYS

API

bridge.data.collators.visual.THW_GRID_VISUAL_KEYS

(‘pixel_values’, ‘pixel_values_videos’, ‘image_grid_thw’, ‘video_grid_thw’)

bridge.data.collators.visual.PASSTHROUGH_VISUAL_KEYS

(‘pixel_values’, ‘pixel_values_videos’, ‘image_grid_thw’, ‘video_grid_thw’, ‘image_sizes’, ‘image_po…