src/transformers/models/groupvit/modeling_groupvit.py/GroupViTMixerMLP/forward
class GroupViTMixerMLP:    def forward(self, x):
        x = super().forward(x.transpose(1, 2))
        return x.transpose(1, 2)