Module diffusion 

Diffusion Models (Stable Diffusion, DDPM, DDIM)

Implements diffusion models for image generation:

• DDPM (Denoising Diffusion Probabilistic Models)
• DDIM (Denoising Diffusion Implicit Models)
• Stable Diffusion architecture
• Noise scheduling
• Sampling algorithms

Structs

DDPM

DDPM (Denoising Diffusion Probabilistic Model)

DiffusionConfig

Diffusion model configuration

NoiseScheduler

Noise scheduler

ResidualBlock

Residual block for U-Net

UNet

U-Net for diffusion models

Enums

BetaSchedule

Beta schedule for noise