src/transformers/models/xlm_roberta/modeling_flax_xlm_roberta.py/FlaxXLMRobertaModule/setup
class FlaxXLMRobertaModule:    def setup(self):
        self.embeddings = FlaxXLMRobertaEmbeddings(self.config, dtype=self.dtype)
        self.encoder = FlaxXLMRobertaEncoder(
            self.config,
            dtype=self.dtype,
            gradient_checkpointing=self.gradient_checkpointing,
        )
        self.pooler = FlaxXLMRobertaPooler(self.config, dtype=self.dtype)