megatron.utils.unwrap_model#

megatron.utils.unwrap_model(model, module_instances=<class 'torch.nn.parallel.distributed.DistributedDataParallel'>)#