code: def get_deepspeed_model(model):
      ds_config_dict = {
          "train_micro_batch_size_per_gpu": 1,
          "optimizer": {"type": "Lamb", "params": {"lr": 0.00015}},
      }
  
      from megatron import mpu
  
      model, _, _, _ = deepspeed.initialize(
          model=model,
          mpu=mpu,
          model_parameters=model.parameters(),
          config=ds_config_dict,
      )
      return model
  
imports: import os
  import torch
  import deepspeed
  import pytest
  import random
  import numpy
  import deepspeed.comm
  from deepspeed.accelerator import get_accelerator
  from unit.common import DistributedTest, DistributedFixture
  from unit.megatron_model import get_gpt2_model, get_megatron_version
  from deepspeed.runtime.utils import required_torch_version
label_desc: Use image node_id 1 for calling a global function with example usage: get_deepspeed_model(model) and returns: model