class_docstr: Shard the original huggingface model according to the policy
  
  Args:
      policy (:class:`Policy`): The policy to shard the model
      model (:class:`torch.Module`): The model to shard
      shard_config: The setting of distributed model
docstr: Replace the parameter of the layer
  
  Args:
      module (:class:`torch.nn.Module`): The object of layer to shard
      param_replacement (List[Callable]): The function list to get parameter shard information in policy
docstr_tok: ["Replace","the","parameter","of","the","layer","Args",":","module","(",":","class",":","`","torch.nn.Module","`",")",":","The","object","of","layer","to","shard","param_replacement","(","List","[","Callable","]",")",":","The","function","list","to","get","parameter","shard","information","in","policy"]
code: def _replace_param(
      self,
      module: nn.Module,
      param_replacement: List[Callable],
  ) -> None:
      r"""
      Replace the parameter of the layer
  
      Args:
          module (:class:`torch.nn.Module`): The object of layer to shard
          param_replacement (List[Callable]): The function list to get parameter shard information in policy
      """
      for param_func in param_replacement:
          param_func(module)
  
imports: from types import MethodType
  from typing import Any, Callable, Dict, List, Optional, Set, Union
  import torch.nn
  from torch import Tensor
  from colossalai.lazy import LazyInitContext
  from .._utils import getattr_, setattr_
  from ..policies.auto_policy import get_autopolicy
  from ..policies.base_policy import Policy, SubModuleReplacementDescription
  from .shard_config import ShardConfig
  from .utils import set_tensors_to_none
label_desc: Use image node_id 8 for calling the ModelSharder obj's underlying member method code with example usage: obj._replace_param(module, param_replacement) without return types