src/transformers/models/groupvit/modeling_tf_groupvit.py/TFGroupViTStage/build
class TFGroupViTStage:    def build(self, input_shape: tf.TensorShape):
        if self.num_group_token > 0:
            self.group_token = self.add_weight(
                shape=(1, self.num_group_token, self.config.hidden_size),
                initializer="zeros",
                trainable=True,
                name="group_token",
            )
        else:
            self.group_token = None
        super().build(input_shape)