src/transformers/models/albert/modeling_albert.py/AlbertLayerGroup/__init__
class AlbertLayerGroup:    def __init__(self, config: AlbertConfig):
        super().__init__()

        self.albert_layers = nn.ModuleList([AlbertLayer(config) for _ in range(config.inner_group_num)])