src/transformers/models/flava/modeling_flava.py/FlavaImageCodebookBlock/forward
class FlavaImageCodebookBlock:    def forward(self, x: torch.Tensor) -> torch.Tensor:
        return self.id_path(x) + self.post_gain * self.res_path(x)