Compute the entire variational lower-bound, measured in bits-per-dim, as well as other related quantities. :param model: the

Center cropping implementation from ADM. https://github.com/openai/guided-diffusion/blob/8fb3ad9197f16bbc40620447b2742e13458d2831/guided_diff

Center cropping implementation from ADM. https://github.com/openai/guided-diffusion/blob/8fb3ad9197f16bbc40620447b2742e13458d2831/guided_diff

Compute the log-likelihood of a continuous Gaussian distribution. :param x: the targets :param means: the Gaussian mean Tensor. :para

Create a logger that writes to a log file and stdout.

Create a ScheduleSampler from a library of pre-defined samplers. :param name: the name of the sampler. :param diffusion: the diffusion ob

Sample x_{t+1} from the model using DDIM reverse ODE.

Generate samples from the model using DDIM. Same usage as p_sample_loop().

Forward pass of DiT, but also batches the unconditional forward pass for classifier-free guidance.

Forward pass of DiT, but also batches the unconditional forward pass for classifier-free guidance.

Get a pre-defined beta schedule for the given name. The beta schedule library consists of beta schedules which remain similar in the limi

Set requires_grad flag for all parameters in a model.

Compute training losses for a single timestep. :param model: the model to evaluate loss on. :param x_start: the [N x C x ...]

Step the EMA model towards the current model.

Update the reweighting using losses from a model. Call this method from each rank with a batch of timesteps and the correspon