bridge.data.collators.registry

Lazy resolution of model-owned VLM collators by HF processor type.

Module Contents

Classes

_ModelCollateSpec

Functions

model_collate_required_for_all_examples	Return whether a processor must always use its model-owned collator.
resolve_model_collate	Resolve a model-owned collator without importing unrelated model modules.

Data

_MODEL_COLLATE_SPECS

API

class bridge.data.collators.registry._ModelCollateSpec

module_name: str

None

symbol_name: str

None

required_for_all_examples: bool

False

bridge.data.collators.registry._MODEL_COLLATE_SPECS

None

bridge.data.collators.registry.model_collate_required_for_all_examples(processor_type: str) → bool

Return whether a processor must always use its model-owned collator.

bridge.data.collators.registry.resolve_model_collate(

processor_type: str,

) → collections.abc.Callable[..., dict[str, Any]]

Resolve a model-owned collator without importing unrelated model modules.