starling.models.continuous_diffusion.karras_log_snr

karras_log_snr(t, sigma_min=0.002, sigma_max=80.0, rho=7.0)

Implements the noise schedule from Karras et al. (2022) “Elucidating the Design Space of Diffusion-Based Generative Models”