diffusion_models.gaussian_diffusion.base_diffuser

Module Contents

class BaseDiffuser(beta_scheduler)

Initializes the object with the specified beta scheduler.

BaseDiffuser is an abstract base class for different diffuser implementations. It defines the interface that all diffusers should adhere to.

Parameters:

beta_scheduler (BaseBetaScheduler) – The beta scheduler used by the diffuser.

Warning

Do not instantiate this class directly. Instead, build your own Diffuser by inheriting from BaseDiffuser. (see GaussianDiffuser)

beta_scheduler: BaseBetaScheduler

The beta scheduler used by the diffuser.

abstract diffuse_batch(images)

Diffuse a batch of images.

Parameters:

images (Tensor) – A tensor containing a batch of images.

Returns:

A tuple containing three tensors

• images: Diffused batch of images.

• noise: Noise added to the images.

• timesteps: Timesteps used for diffusion.

Return type:

Tuple[Tensor, Tensor, Tensor]

abstract denoise_batch(images, model)

Denoise a batch of images.

Parameters:

• images (Tensor) – A tensor containing a batch of images to denoise.

• model (diffusion_models.models.base_diffusion_model.BaseDiffusionModel) – The model to be used for denoising.

Returns:

A list of tensors containing a batch of denoised images.

Return type:

List[Tensor]

abstract to(device='cpu')

Moves the data to the specified device.

This performs a similar behaviour to the to method of PyTorch.

Parameters:

device (str) – The device to which the method should move the data. It should be a string indicating the desired device. Default is “cpu”.