code: def get_model(args):
      config = CPMBeeConfig.from_json_file(args.model_config)
      model = CPMBee(config)
      if args.load is not None:
          bmt.load(model, args.load)
      else:
          bmt.init_parameters(model)
      return model
  
imports: import json
  import time
  from typing import Any, Dict, List, Union
  import torch
  import bmtrain
  import os
  from cpm_live.arguments import get_args
  from cpm_live.models import CPMBee, CPMBeeConfig
  from cpm_live.tokenizers import CPMBeeTokenizer
  from cpm_live.utils import allgather_objects, LogManager
  from cpm_live.training_tasks.bee import MixedDataset
label_desc: Use image node_id 2 for calling a global function with example usage: get_model(args) and returns: model