Method forward

k_diffusion/layers.py:259–260

(self, input)

        self.skip = skip if skip else nn.Identity()

    def forward(self, input):
        return self.main(input) + self.skip(input)


# Noise level (and other) conditioning