src/transformers/models/clip/modeling_flax_clip.py/FlaxCLIPTextModule/setup
class FlaxCLIPTextModule:    def setup(self):
        self.text_model = FlaxCLIPTextTransformer(self.config, dtype=self.dtype)