diffusion_models.gaussian_diffusion.base_diffuser

Module Contents

class BaseDiffuser(beta_scheduler)[source]

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

beta_scheduler[source]
abstract diffuse_batch(images)[source]
abstract denoise_batch(images, model)[source]
abstract to(device='cpu')[source]