src/transformers/models/groupvit/modeling_groupvit.py/GroupViTStage/split_x
class GroupViTStage:    def split_x(self, x):
        if self.with_group_token:
            return x[:, : -self.num_group_token], x[:, -self.num_group_token :]
        else:
            return x, None