bridge.data.collators.registry

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

Module Contents

Classes

_ModelCollateSpec

Functions

always_use_model_collate	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

always_use_model_collate: bool

False

bridge.data.collators.registry._MODEL_COLLATE_SPECS

None

bridge.data.collators.registry.always_use_model_collate(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.