The guided-diffusion repository is centered on diffusion models for image synthesis, with a focus on classifier guidance and improvements over earlier diffusion frameworks. It is derived from OpenAI’s improved-diffusion work, enhanced to include guided generation where a classifier (or other guidance mechanism) can steer sampling toward desired classes or attributes. The code provides model definitions (UNet, diffusion schedules), sampling and training scripts, and utilities for guidance and evaluation. A key insight is that combining diffusion sampling with classifier gradients allows fine control over the generated images, trading off diversity vs fidelity. The repository includes scripts such as image_train.py, image_sample.py, and classifier_train.py to train diffusion models, generate samples, and train guiding classifiers. It also ships with precomputed evaluation batches and baseline comparisons to support reproducible benchmarking of new models.

Features

  • Diffusion model architecture (UNet, noise schedules, training utilities)
  • Classifier-guided sampling: combining diffusion with classifier gradients
  • Scripts for training models (image_train.py), sampling (image_sample.py), and classifier training
  • Precomputed evaluation batches and baseline metrics for reproducibility
  • Modular code enabling new guidance modalities or architectural tweaks
  • Branching from improved-diffusion with enhancements in guided generation

Project Samples

Project Activity

See All Activity >

License

MIT License

Follow Guided Diffusion

Guided Diffusion Web Site

Other Useful Business Software
Gen AI apps are built with MongoDB Atlas Icon
Gen AI apps are built with MongoDB Atlas

The database for AI-powered applications.

MongoDB Atlas is the developer-friendly database used to build, scale, and run gen AI and LLM-powered apps—without needing a separate vector database. Atlas offers built-in vector search, global availability across 115+ regions, and flexible document modeling. Start building AI apps faster, all in one place.
Start Free
Rate This Project
Login To Rate This Project

User Reviews

Be the first to post a review of Guided Diffusion!

Additional Project Details

Programming Language

Python

Related Categories

Python Artificial Intelligence Software

Registered

11 hours ago