diffusion_models.gaussian_diffusion.gaussian_diffuser

Module Contents

class GaussianDiffuser(beta_scheduler)[source]

Helper class that provides a standard way to create an ABC using inheritance.

device = 'cuda'[source]
classmethod from_checkpoint(checkpoint)[source]
to(device='cpu')[source]
diffuse_batch(images)[source]
denoise_batch(images, model)[source]