code: def _choose_module_key(self, sd):
      assert not (
          "module" in sd and "model" in sd
      ), "checkpoint has both 'model' and 'module' keys, not sure how to proceed"
      assert (
          "module" in sd or "model" in sd
      ), "checkpoint contains neither 'model' or 'module' keys, not sure how to proceed"
      if "module" in sd:
          return "module"
      elif "model" in sd:
          return "model"
  
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 5 for calling the SDLoaderBase obj's underlying member method code with example usage: obj._choose_module_key(sd) and returns: str, str