`bridge.models.muse_spark_open.muse_spark_open_builder`#

Builder for Muse Spark Open pipeline stages.

Module Contents#

Classes#

MuseSparkModelBuilder

Build the decoder and HF perception modules in a single model stage.

API#

class bridge.models.muse_spark_open.muse_spark_open_builder.MuseSparkModelBuilder#

Bases: megatron.training.models.hybrid.HybridModelBuilder

Build the decoder and HF perception modules in a single model stage.

build_model( pg_collection: megatron.core.process_groups_config.ProcessGroupCollection, pre_process: bool | None = None, post_process: bool | None = None, vp_stage: int | None = None, ) → megatron.bridge.models.muse_spark_open.modeling_muse_spark_open.model.MuseSparkModel#: Construct the requested pipeline stage from the imported config.