src/transformers/models/yolos/modeling_yolos.py/YolosAttention/__init__
class YolosAttention:    def __init__(self, config: YolosConfig) -> None:
        super().__init__()
        self.attention = YolosSelfAttention(config)
        self.output = YolosSelfOutput(config)
        self.pruned_heads = set()