megatron.training.get_model

megatron.training.get_model(model_provider_func: Callable, model_type=ModelType.encoder_or_decoder, wrap_with_ddp: bool = True, args=None)

Build the model.