code: def sanity_check(self, ckpt_file_name):
      keys_to_check = [
          "attention.dense.weight",
          "mlp.dense_4h_to_h.weight",
          "attention.query_key_value",
          "mlp.dense_h_to_4h.weight",
          "mlp.dense_h_to_4h.bias",
      ]
  
      sd = self.checkpoint_engine.load(
          ckpt_file_name, map_location=lambda storage, loc: storage
      )
  
      # partial_key is a sub-string of one key in the sd
      def check_key_exist(partial_key, sd):
          keys = sd.keys()
          found = False
          for k in keys:
              if partial_key in k:
                  found = True
                  break
          return found
  
      for key in keys_to_check:
          assert check_key_exist(
              key, self.get_module(sd)
          ), f"key: {key} is not found in the checkpoint {ckpt_file_name}"
  
imports: import torch
  import os
  import copy
  import collections
  import json
  from abc import ABC, abstractmethod
  from deepspeed.utils import logger
  from deepspeed.runtime.checkpoint_engine.torch_checkpoint_engine import TorchCheckpointEngine
  from .weight_quantizer import WeightQuantization
label_desc: Use image node_id 6 for calling the MegatronSDLoader obj's underlying member method code with example usage: obj.sanity_check(ckpt_file_name) and returns: found