external_model_hub

Model hub for loading external models.

This module contains mappings and callbacks for external model checkpoints used in the Lightning IR library.

lightning_ir.base.external_model_hub.CHECKPOINT_MAPPING

Mapping of model checkpoint identifiers to their configurations.

Type:

Dict[str, LightningIRConfig]

lightning_ir.base.external_model_hub.STATE_DICT_KEY_MAPPING

Mapping of state dictionary keys for model checkpoints.

Type:

Dict[str, List[Tuple[str | None, str]]]

lightning_ir.base.external_model_hub.POST_LOAD_CALLBACKS

Callbacks to be executed after loading a model.

Type:

Dict[str, Callable[[LightningIRModel], LightningIRModel]]