src/transformers/models/videomae/modeling_videomae.py/VideoMAEAttention/__init__
class VideoMAEAttention:    def __init__(self, config: VideoMAEConfig) -> None:
        super().__init__()
        self.attention = VideoMAESelfAttention(config)
        self.output = VideoMAESelfOutput(config)
        self.pruned_heads = set()