code: def get_model_ipu(
      model_name: str, token: Optional[str] = None
  ) -> VLM_IPU:
      config_path, state, tokenizer_path = get_checkpoint(
          model_name, token
      )
  
      with open(config_path, "r") as f:
          model = VLM_IPU(load(f), tokenizer_path)
  
      model.image_encoder.load_state_dict(state["image_encoder"])
      model.text_encoder.load_state_dict(state["text_encoder"])
  
      return model.eval()
  
file_path: uform/src/uform/__init__.py
imports: from json import load
  from typing import Mapping, Optional, Tuple
  import torch
  from huggingface_hub import snapshot_download
  from uform.models import *
label_desc: Use image node_id 4 for calling a global function with example usage: get_model_ipu(model_name, token) and returns: model