src/transformers/models/groupvit/modeling_groupvit.py/GroupViTCrossAttentionLayer/forward
class GroupViTCrossAttentionLayer:    def forward(self, query, key):
        x = query
        x = x + self.attn(query, encoder_hidden_states=key)[0]
        x = x + self.mlp(self.norm2(x))
        x = self.norm_post(x)
        return x