NoiseScheduler#

class tfts.models.diffusion.NoiseScheduler(config: DiffusionConfig)[source]#

Bases: object

Linear noise scheduler for diffusion models

Inherited-members:

Methods

add_noise(x, t)

Add noise to the input at timestep t

remove_noise(x, noise, t)

Remove noise from the input at timestep t
add_noise(x: Tensor, t: Tensor) → Tuple[Tensor, Tensor][source]#

Add noise to the input at timestep t

remove_noise(x: Tensor, noise: Tensor, t: Tensor) → Tensor[source]#

Remove noise from the input at timestep t