src/transformers/models/xlm_roberta/modeling_flax_xlm_roberta.py/FlaxXLMRobertaAttention/setup
class FlaxXLMRobertaAttention:    def setup(self):
        self.self = FlaxXLMRobertaSelfAttention(self.config, causal=self.causal, dtype=self.dtype)
        self.output = FlaxXLMRobertaSelfOutput(self.config, dtype=self.dtype)