code: def _load(input_file):
      import torch
      from omegaconf import OmegaConf
  
      data = torch.load(input_file)
      cfg = OmegaConf.to_container(data["cfg"])
      for key in list(cfg.keys()):
          if key != "model":
              del cfg[key]
              if "w2v_args" in cfg["model"]:
                  del cfg["model"]["w2v_args"][key]
      state_dict = {
          _removeprefix(k, "w2v_encoder."): v
          for k, v in data["model"].items()
      }
      return cfg, state_dict
  
label_desc: Use image node_id 3 for calling a global function with example usage: _load(input_file) and returns: cfg,  state_dict