src/transformers/models/longt5/modeling_flax_longt5.py/FlaxLongT5LayerCollection/setup
class FlaxLongT5LayerCollection:    def setup(self):
        self.layer = FlaxLongT5Block(
            self.config, has_relative_attention_bias=self.has_relative_attention_bias, dtype=self.dtype
        )