Method __init__

improved_diffusion/unet.py:533–534

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

    """

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

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