code: def set_mlp_state(args, layer, hf_layer):
      """Set MLP params."""
  
      mlp = layer.mlp
      hf_mlp = hf_layer.mlp
  
      mlp.dense_h_to_4h.weight.data.copy_(
          torch.cat(
              [
                  hf_mlp.gate_proj.weight,
                  hf_mlp.up_proj.weight,
              ],
              dim=0,
          )
      )
      mlp.dense_4h_to_h.weight.data.copy_(hf_mlp.down_proj.weight)
  
file_path: megatron-lm/tools/checkpoint/loader_llama2_hf.py
imports: import json
  import os
  import sys
  import torch
  import transformers
  from tqdm import tqdm
  import types
label_desc: Use image node_id 7 for calling a global function with example usage: set_mlp_state(args, layer, hf_layer) without return types