`bridge.models.common.base`#

Module Contents#

Classes#

Bridge compatibility wrapper for Megatron-LM model configs.

API#

class bridge.models.common.base.ModelConfig#

Bases: megatron.training.models.base.ModelConfig

Bridge compatibility wrapper for Megatron-LM model configs.

get_builder_cls() → type#: Get the appropriate builder type for this config.

classmethod from_dict( data: dict[str, Any], ) → bridge.models.common.base.ModelConfig#: Deserialize config from dictionary with Bridge target validation.