code: def load_checkpoint(filepath, device):
      assert os.path.isfile(filepath)
      print("Loading '{}'".format(filepath))
      checkpoint_dict = torch.load(filepath, map_location=device)
      print("Complete.")
      return checkpoint_dict
  
file_path: underthesea/underthesea/pipeline/say/viettts_/hifigan/convert_torch_model_to_haiku.py
imports: import argparse
  import json
  import os
  import pickle
  import numpy
  import torch
  from .config import FLAGS
  from .torch_model import Generator
label_desc: Use image node_id 1 for calling a global function with example usage: load_checkpoint(filepath, device) and returns: checkpoint_dict