model_support.registry

model_support.registry

Registry mapping model_type to its ModelSupport descriptor.

Functions

Name Description
get_model_support Look up the descriptor for a model_type; None if unregistered.
get_model_support_for_cfg Look up a descriptor before the model config is loaded.
get_model_support_for_processor Look up a descriptor by multimodal processor instance.
register_model_support Class decorator registering a descriptor under each of its model_types.

get_model_support

model_support.registry.get_model_support(model_type)

Look up the descriptor for a model_type; None if unregistered.

get_model_support_for_cfg

model_support.registry.get_model_support_for_cfg(cfg)

Look up a descriptor before the model config is loaded.

An exact resolved model_config_type wins. Before that is available, descriptors can match on config fields such as the model name.

get_model_support_for_processor

model_support.registry.get_model_support_for_processor(processor)

Look up a descriptor by multimodal processor instance.

register_model_support

model_support.registry.register_model_support(support_cls)

Class decorator registering a descriptor under each of its model_types.

Out-of-tree architectures can call this from a plugin or any imported module; registering an already-covered model_type overrides the built-in descriptor.