code: def set_postprocess_state(args, model, hf_model):
      """Set output layer & norm params."""
      model.language_model.encoder.final_norm.weight.data.copy_(
          hf_model.model.norm.weight
      )
      model.language_model.output_layer.weight.data.copy_(
          hf_model.lm_head.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 5 for calling a global function with example usage: set_postprocess_state(args, model, hf_model) without return types