Method _weighting_edm2

k_diffusion/layers.py:80–81

(self, sigma)

        return self.sigma_data ** 2 / (sigma ** 2 + self.sigma_data ** 2)

    def _weighting_edm2(self, sigma):
        return (sigma ** 2 + self.sigma_data ** 2) / (sigma * self.sigma_data) ** 2

    def get_scalings(self, sigma):
        c_skip = self.sigma_data ** 2 / (sigma ** 2 + self.sigma_data ** 2)