src/transformers/models/wav2vec2/modeling_flax_wav2vec2.py/FlaxWav2Vec2PositionalConvEmbedding/setup
class FlaxWav2Vec2PositionalConvEmbedding:    def setup(self):
        self.conv = FlaxConvWithWeightNorm(self.config, dtype=self.dtype)
        self.activation = ACT2FN[self.config.feat_extract_activation]
        self.num_pad_remove = 1 if self.config.num_conv_pos_embeddings % 2 == 0 else 0