def METHOD_NAME(self, module, sd, weight_quantizer, mp_replace, prefix):
    param_names = (
        'self_attn.q_proj.weight', \
        'self_attn.k_proj.weight', \
        'self_attn.v_proj.weight', \
        'self_attn.o_proj.weight', \
        'mlp.up_proj.weight', \
        'mlp.gate_proj.weight', \
        'mlp.down_proj.weight', \
        'input_layernorm.weight', \
        'post_attention_layernorm.weight'
    )
    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])
    maybe_copy_geglu(module.mlp, sd, weight_quantizer, mp_replace, 'inter_w',
                     [prefix + param_names[4], prefix + param_names[5]])
    maybe_copy(module.mlp, sd, weight_quantizer, mp_replace, 'output_w', prefix + param_names[6])
    maybe_copy(module.mlp, sd, weight_quantizer, mp_replace, transformer_param_names[8], prefix + param_names[7])
    maybe_copy(module, sd, weight_quantizer, mp_replace, transformer_param_names[10], prefix + param_names[8])