code: def setUp(self):
      config = clip_b32.get_config(init_mode="canonical_checkpoint")
      config.dataset_configs.input_size = 128
      module = models.TextZeroShotDetectionModule(
          body_configs=config.model.body,
          normalize=config.model.normalize,
          box_bias=config.model.box_bias,
      )
      rng = jax.random.PRNGKey(0)
      image = jnp.zeros((1, 128, 128, 3), dtype=jnp.float32)
      text_queries = jnp.zeros(
          (1, 5, config.dataset_configs.max_query_length),
          dtype=jnp.int32,
      )
      variables = module.init(rng, image, text_queries, train=False)
      self.model = inference.Model(config, module, variables)
      self.num_instances = (
          config.dataset_configs.input_size // 32
      ) ** 2
      super().setUp()
  
      self.enter_context(
          mock.patch.object(
              target=models.clip_tokenizer,
              attribute="tokenize",
              autospec=True,
              side_effect=_mock_tokenize,
          )
      )
  
file_path: scenic/scenic/projects/owl_vit/notebooks/tests/inference_test.py
imports: from unittest import mock
  from absl.testing import absltest
  import chex
  import jax
  import jax.numpy
  import numpy
  from scenic.projects.owl_vit import models
  from scenic.projects.owl_vit.configs import clip_b32
  from scenic.projects.owl_vit.notebooks import inference
label_desc: Use image node_id 1 for calling the InferenceTest obj's underlying member method code with example usage: obj.setUp() without return types