nami.schedules.ve# Classes VESchedule([sigma_min, sigma_max]) class nami.schedules.ve.VESchedule(sigma_min=0.01, sigma_max=50.0)[source]# Bases: NoiseSchedule Parameters: sigma_min (float) sigma_max (float) alpha(t)[source]# Return type: Tensor Parameters: t (Tensor) diffusion(t)[source]# Return type: Tensor Parameters: t (Tensor) drift(x, t)[source]# Return type: Tensor Parameters: x (Tensor) t (Tensor) sigma(t)[source]# Return type: Tensor Parameters: t (Tensor)
