code: def __init__(
      self, img_model_type="uniter-base", txt_model_type="bert-base"
  ):
      self.img_model_type = img_model_type
      self.txt_model_type = txt_model_type
      self.img_model_config = "img_model_config"
      self.txt_model_config = "txt_model_config"
      self.img_checkpoint = "./"
      self.txt_checkpoint = "./"
  
file_path: towhee/tests/unittests/models/lightning_dot/test_lightning_dot.py
imports: import unittest
  import torch
  from torch import nn
  from towhee.models.lightning_dot.bi_encoder import BiEncoder
label_desc: Use image node_id 1 to create a new MockArgs object  with example: obj = MockArgs(img_model_type, txt_model_type)