code: def load_model(model_name, quantization):
      model = LlamaForCausalLM.from_pretrained(
          model_name,
          return_dict=True,
          load_in_8bit=quantization,
          device_map="auto",
          low_cpu_mem_usage=True,
      )
      return model
  
imports: from peft import PeftModel
  from transformers import LlamaForCausalLM, LlamaConfig
label_desc: Use image node_id 1 for calling a global function with example usage: load_model(model_name, quantization) and returns: model