src/transformers/models/wav2vec2/modeling_flax_wav2vec2.py/FlaxWav2Vec2AdapterLayer/setup
class FlaxWav2Vec2AdapterLayer:    def setup(self):
        self.conv = nn.Conv(
            features=2 * self.config.output_hidden_size,
            kernel_size=(self.config.adapter_kernel_size,),
            strides=(self.config.adapter_stride,),
            padding=((1, 1),),
            kernel_init=jax.nn.initializers.normal(self.config.initializer_range),
            dtype=self.dtype,
        )