src/transformers/models/owlvit/modeling_owlvit.py/OwlViTBoxPredictionHead/__init__
class OwlViTBoxPredictionHead:    def __init__(self, config: OwlViTConfig):
        super().__init__()

        width = config.vision_config.hidden_size
        self.dense0 = nn.Linear(width, width)
        self.dense1 = nn.Linear(width, width)
        self.gelu = nn.GELU()
        self.dense2 = nn.Linear(width, 4)