Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
lqiang67 authored Dec 16, 2024
1 parent e54c261 commit 68edff0
Showing 1 changed file with 8 additions and 11 deletions.
19 changes: 8 additions & 11 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -15,21 +15,18 @@ ______________________________________________________________________

## Overview

**RectifiedFlow** provides a *unified* and *minimal* PyTorch codebase for diffusion and flow models. By leveraging a simplified perspective from Rectified Flow, it delivers a streamlined and user-friendly platform for training and inference. The design prioritizes simplicity, intuitive usage, and rapid prototyping, while also supporting state-of-the-art model training and inference. The library includes:
**RectifiedFlow** is a simple, unified PyTorch codebase for diffusion and flow models. It offers an easy-to-use platform for training and inference, focusing on simplicity, flexibility, and quick prototyping. The library includes:

- **Companion Resources**: Includes a [![Blog](https://img.shields.io/badge/blog-blue)](https://rectifiedflow.github.io), [![Lecture Notes](https://img.shields.io/badge/lecture%20notes-blue)](https://github.com/lqiang67/rectified-flow/tree/main/pdf), and beginner-friendly [![Notebooks](https://img.shields.io/badge/Notebooks-orange)](https://github.com/lqiang67/rectified-flow/tree/main/examples) covering concepts from basics to advanced implementations.

- **Companion Resources**: Accompanied by a matrix of learning materials, including [![Blog](https://img.shields.io/badge/blog-blue)](https://rectifiedflow.github.io) and [![Lecture Notes](https://img.shields.io/badge/lecture%20notes-blue)](https://github.com/lqiang67/rectified-flow/tree/main/pdf), as well as beginner-friendly [![Notebooks](https://img.shields.io/badge/Notebooks-orange)](https://github.com/lqiang67/rectified-flow/tree/main/examples) that provide comprehensive guidance from basic concepts to advanced implementations.
- **Simplified ODE Perspective**: Train and infer rectified flow (RF) and diffusion models using a unified ODE approach, including learning 1-rectified flow from data (a.k.a. flow matching), reflow for speedup, and diffusion as stochastic RF sampling.

- **Simplified ODE Perspective**: Seamlessly train and infer rectified flow (RF) and diffusion models from a single, coherent ODE perspective. It includes learning 1-rectified flow from data (a.k.a. flow matching), and reflow for speed up, and diffusion models as stochastic samplers of RF.
- **Easy Integration with SOTA Models**: Easily integrate state-of-the-art models, including the Flux series, for greater flexibility and compatibility.

- **Easy Integration with SOTA Models**: Effortlessly integrate state-of-the-art models, including Flux series, into the framework for enhanced flexibility and compatibility.

- **Comprehensive Tools**:
- **Symbolic Interpolation Solvers**: Automatically handles affine interpolation, including interpolation and derivative computations, and provides symbolic solutions for interpolation equations.
- **Model Form Interconversion**: Seamlessly converts between different model forms, including score models, velocity predictions, and noise / image predictions.
- **Deterministic & Stochastic Sampling**: Offers unified support for both deterministic and stochastic sampling methods, enabling the easy implementation of various algorithmic approaches (such as DDIM and DDPM) within a single cohesive framework.

Whether you are a researcher exploring the frontiers of generative modeling, a student seeking to deepen your understanding through comprehensive tutorials, or a scientist investigating state-of-the-art text-to-image generation, **RectifiedFlow** offers the essential resources and functionalities to advance your projects with confidence and ease.
- **Comprehensive Tools**:
- **Symbolic Interpolation**: Automates affine interpolation and its derivatives with symbolic solutions.
- **Model Conversion**: Converts between score models, velocity predictions, and noise/image predictions.
- **Stochastic Sampling**: Supports both deterministic and stochastic sampling (e.g., DDPM) in one framework.

---

Expand Down

0 comments on commit 68edff0

Please sign in to comment.