code: def build_model(name="vit_b_16x"):
      assert (
          name in MODEL_NAMES
      ), f"model name must be one of {MODEL_NAMES}"
      name2model = {"vit_b_16x": build_vit_b_16x_model}
      model = name2model[name]()
      return model
  
file_path: PaddleHub/modules/image/text_to_image/disco_diffusion_ernievil_base/vit_b_16x/ernievil2/utils/utils.py
imports: import json
  import os
  from typing import List
  from typing import Union
  import numpy
  import paddle
  from disco_diffusion_ernievil_base.vit_b_16x.ernievil2.transformers.clip_vision_transformer import ViT_base_patch16_224
  from disco_diffusion_ernievil_base.vit_b_16x.ernievil2.transformers.clip_vision_transformer import ViT_base_patch32_224
  from disco_diffusion_ernievil_base.vit_b_16x.ernievil2.transformers.clip_vision_transformer import ViT_large_patch14_224
  from disco_diffusion_ernievil_base.vit_b_16x.ernievil2.transformers.efficientnet import EfficientNetB5
  from disco_diffusion_ernievil_base.vit_b_16x.ernievil2.transformers.ernie2 import ErnieModel
  from disco_diffusion_ernievil_base.vit_b_16x.ernievil2.transformers.multimodal import MultiModalModel
  from disco_diffusion_ernievil_base.vit_b_16x.ernievil2.utils.tokenizer import FullTokenizer
label_desc: Use image node_id 2 for calling a global function with example usage: build_model(name) and returns: model