src/transformers/models/wav2vec2/modeling_flax_wav2vec2.py/FlaxWav2Vec2FeatureEncoder/setup
class FlaxWav2Vec2FeatureEncoder:    def setup(self):
        self.conv_layers = FlaxConvLayersCollection(self.config, dtype=self.dtype)