autocast.processors.noise

class NoiseSchedule(*args, **kwargs)

Bases: Module

Noise Schedule Module.

if Azula became useful for other things we can use their base class instead of this base

Parameters:

• args (Any)

• kwargs (Any)

forward(t)

Get alpha and sigma for given time steps t.

Parameters:

t (Tensor)

Return type:

tuple[Tensor, Tensor]

class LogLinearSchedule(sigma_min=0.002, sigma_max=80.0)

Bases: NoiseSchedule

Log-Linear Noise Schedule.

Implements a log-linear schedule for alpha and sigma.

Parameters:

• sigma_min (float)

• sigma_max (float)

forward(t)

Get alpha and sigma for given time steps t.

Parameters:

t (Tensor)

Return type:

tuple[Tensor, Tensor]