Diffusion model with a Transformer backbone.

A DiT block with adaptive layer norm zero (adaLN-Zero) conditioning.

Calculate pairwise distances between vectors. Adapted from https://github.com/kynkaat/improved-precision-and-recall-metric/blob/f60f25e5ad93

The final layer of DiT.

The final layer of DiT.

Utilities for training and sampling diffusion models. Original ported from this codebase: https://github.com/hojonathanho/diffusion/blob/

Embeds class labels into vector representations. Also handles label dropout for classifier-free guidance.

Embeds class labels into vector representations. Also handles label dropout for classifier-free guidance.

A helper for comparing manifolds of feature vectors. Adapted from https://github.com/kynkaat/improved-precision-and-recall-metric/blob/f60f2

A diffusion process which can skip steps in a base diffusion process. :param use_timesteps: a collection (sequence or set) of timesteps from

Embeds scalar timesteps into vector representations.

Embeds scalar timesteps into vector representations.

DIFFUSION CONVOLUTION MODEL: DiC.

Which type of output the model predicts.

What is used as the model's output variance. The LEARNED_RANGE option has been added to allow the model to predict values between FIXED_S

A distribution over timesteps in the diffusion process, intended to reduce variance of the objective. By default, samplers perform unbias