code: def load_model(model_name, tp_size=1):
      llm = LLM(model_name, tensor_parallel_size=tp_size)
      return llm
  
file_path: llama-recipes/examples/vllm/inference.py
imports: import fire
  import torch
  from vllm import LLM
  from vllm import LLM, SamplingParams
  from accelerate.utils import is_xpu_available
label_desc: Use image node_id 1 for calling a global function with example usage: load_model(model_name, tp_size) and returns: llm