code: def __init__(self, config):
      check_vit_in_transformers()
      super().__init__(config)
      # We need to support attention masks for vision language input
      # ViTAttention from transformers doesn't currently support attention masks,
      # for versions without attention_mask support we use these clones of ViT modules
      # that use BertSelfAttention to enable masking.
      self.attention = BertSelfAttention(config)
  
imports: from collections import namedtuple
  from dataclasses import asdict, dataclass
  import torch
  from mmf.utils.general import retry_n
  from omegaconf import OmegaConf
  from packaging import version
  from torch import nn
label_desc: Use image node_id 1 to create a new ViTAttention object from inherited base classes: vit with example: obj = ViTAttention(config)