code: def load_params(
      self, module, sd, weight_quantizer, mp_replace, prefix
  ):
      param_names = (
          "attn.q_proj.weight",
          "attn.k_proj.weight",
          "attn.v_proj.weight",
          "attn.out_proj.weight",
          "mlp.fc_in.weight",
          "mlp.fc_in.bias",
          "mlp.fc_out.weight",
          "mlp.fc_out.bias",
          "ln_1.weight",
          "ln_1.bias",
      )
      maybe_copy_qkv(
          module.attention,
          sd,
          weight_quantizer,
          mp_replace,
          "attn_qkvw",
          [
              prefix + param_names[0],
              prefix + param_names[1],
              prefix + param_names[2],
          ],
          split_qkv=self.policy.split_qkv,
      )
      for i in range(3, 4):
          maybe_copy(
              module.attention,
              sd,
              weight_quantizer,
              mp_replace,
              transformer_param_names[i - 1],
              prefix + param_names[i],
          )
      for i in range(4, 8):
          maybe_copy(
              module.mlp,
              sd,
              weight_quantizer,
              mp_replace,
              transformer_param_names[i],
              prefix + param_names[i],
          )
      for i in range(8, 10):
          maybe_copy(
              module,
              sd,
              weight_quantizer,
              mp_replace,
              transformer_param_names[i + 2],
              prefix + param_names[i],
          )
  
imports: from .base import *
  from .features.meta_tensor import MetaTensorContainer
  from .features.split_qkv import HybridSplitQKVContainer
  from deepspeed.model_implementations.transformers.ds_gpt import DeepSpeedGPTInference
  import torch
  from torch.nn.parameter import Parameter
  from ..policy import TransformerPolicy
  from ..policy import transformer_param_names
  from ..policy import maybe_copy
  from ..policy import maybe_copy_qkv
  from ..policy import maybe_get_lora
label_desc: Use image node_id 6 for calling the DS_GPTJContainer obj's underlying member method code with example usage: obj.load_params(module, sd, weight_quantizer, mp_replace, prefix) without return types