code: def setUp(self):
      test_utils.setup_proxy()
      setup_imports()
      model_name = "vilt"
      args = test_utils.dummy_args(model=model_name, dataset="test")
      configuration = Configuration(args)
      config = configuration.get_config()
      model_config = config.model_config[model_name]
      model_config.model = model_name
      self.pretrain_model = build_model(model_config)
  
imports: import gc
  import unittest
  import tests.test_utils
  import torch
  from mmf.common.sample import SampleList
  from mmf.models.vilt import ViLTImageEmbedding, ViLTTextEmbedding
  from mmf.utils.build import build_model
  from mmf.utils.configuration import Configuration
  from mmf.utils.env import setup_imports, teardown_imports
  from mmf.utils.general import get_current_device
  from tests.test_utils import skip_if_old_transformers
  from torch import nn
label_desc: Use image node_id 1 for calling the TestViltPretrained obj's underlying member method code with example usage: obj.setUp() without return types