src/transformers/models/gpt2/modeling_flax_gpt2.py/FlaxGPT2BlockCollection/setup
class FlaxGPT2BlockCollection:    def setup(self):
        self.blocks = [
            FlaxGPT2Block(self.config, name=str(i), dtype=self.dtype) for i in range(self.config.num_hidden_layers)
        ]