docstr: Load model from sharded checkpoint.
code: def load_sharded_model(
      self,
      model: ModelWrapper,
      checkpoint_index_file: str,
      strict: bool = False,
      use_safetensors: bool = False,
      load_sub_module: bool = True,
  ):
      """
      Load model from sharded checkpoint.
      """
      assert isinstance(
          model, ModelWrapper
      ), "Please boost the model before loading!"
      super().load_sharded_model(
          model.unwrap(),
          checkpoint_index_file,
          strict,
          use_safetensors,
          load_sub_module,
      )
  
imports: from typing import Callable, Iterator, List, Optional, Tuple
  import torch.nn
  from torch.nn.parallel import DistributedDataParallel
  from torch.optim import Optimizer
  from torch.optim.lr_scheduler import _LRScheduler
  from torch.utils.data import DataLoader
  from colossalai.checkpoint_io import CheckpointIO, GeneralCheckpointIO
  from colossalai.cluster import DistCoordinator
  from colossalai.interface import ModelWrapper, OptimizerWrapper
  from .dp_plugin_base import DPPluginBase
label_desc: Use image node_id 8 for calling the TorchDDPCheckpointIO obj's underlying member method code with example usage: obj.load_sharded_model(model, checkpoint_index_file, strict, use_safetensors, load_sub_module) without return types