Method __init__

pixel_generator/guided_diffusion/unet.py:684–685

(self, image_size, in_channels, *args, **kwargs)

    """

    def __init__(self, image_size, in_channels, *args, **kwargs):
        super().__init__(image_size, in_channels * 2, *args, **kwargs)

    def forward(self, x, timesteps, low_res=None, **kwargs):
        _, _, new_height, new_width = x.shape