code: def main(base_model: str, peft_model: str, output_dir: str):
      model = LlamaForCausalLM.from_pretrained(
          base_model,
          load_in_8bit=False,
          torch_dtype=torch.float16,
          device_map="auto",
          offload_folder="tmp",
      )
  
      tokenizer = LlamaTokenizer.from_pretrained(base_model)
  
      model = PeftModel.from_pretrained(
          model,
          peft_model,
          torch_dtype=torch.float16,
          device_map="auto",
          offload_folder="tmp",
      )
  
      model = model.merge_and_unload()
      model.save_pretrained(output_dir)
      tokenizer.save_pretrained(output_dir)
  
imports: import fire
  import torch
  from peft import PeftModel
  from transformers import LlamaForCausalLM, LlamaTokenizer
label_desc: Use image node_id 1 for calling a global function with example usage: main(base_model, peft_model, output_dir) without return types