diffusion model

A generative model that learns to reverse a noise-adding process to generate images. Powers Stable Diffusion, DALL-E, and Midjourney.

Syntax

generative-ai

noisy_image -> [denoise steps] -> clean generated image

Example

generative-ai

# Image generation with diffusers:
from diffusers import StableDiffusionPipeline
import torch

pipe = StableDiffusionPipeline.from_pretrained(
    "runwayml/stable-diffusion-v1-5",
    torch_dtype=torch.float16
)
pipe = pipe.to("cuda")

image = pipe(
    "A serene mountain landscape at golden hour, photorealistic"
).images[0]
image.save("output.png")