src/transformers/models/t5/modeling_flax_t5.py/FlaxT5BlockCollection/setup
class FlaxT5BlockCollection:    def setup(self):
        self.causal = self.config.causal
        self.blocks = [
            FlaxT5LayerCollection(self.config, has_relative_attention_bias=(i == 0), dtype=self.dtype, name=str(i))
            for i in range(self.config.num_layers)
        ]