src/transformers/models/t5/modeling_flax_t5.py/FlaxT5LayerCollection/setup
class FlaxT5LayerCollection:    def setup(self):
        self.layer = FlaxT5Block(
            self.config, has_relative_attention_bias=self.has_relative_attention_bias, dtype=self.dtype
        )