code: def set_preprocess_state(args, model, hf_model):
      """Set embedding params."""
      model.language_model.embedding.word_embeddings.weight.data.copy_(
          hf_model.model.embed_tokens.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 4 for calling a global function with example usage: set_preprocess_state(args, model, hf_model) without return types