src/transformers/models/clip/modeling_flax_clip.py/FlaxCLIPTextTransformer/setup
class FlaxCLIPTextTransformer:    def setup(self):
        self.embeddings = FlaxCLIPTextEmbeddings(self.config, dtype=self.dtype)
        self.encoder = FlaxCLIPEncoder(self.config, dtype=self.dtype)
        self.final_layer_norm = nn.LayerNorm(epsilon=self.config.layer_norm_eps, dtype=self.dtype)