src/transformers/models/flava/modeling_flava.py/FlavaImageCodebookLayerGroup/forward
class FlavaImageCodebookLayerGroup:    def forward(self, x: torch.Tensor) -> torch.Tensor:
        return self.group(x)