src/transformers/models/deit/modeling_tf_deit.py/TFDeitPixelShuffle/__init__
class TFDeitPixelShuffle:    def __init__(self, upscale_factor: int, **kwargs) -> None:
        super().__init__(**kwargs)
        if not isinstance(upscale_factor, int) or upscale_factor < 2:
            raise ValueError(f"upscale_factor must be an integer value >= 2 got {upscale_factor}")
        self.upscale_factor = upscale_factor