From 8c16f12fb44b38b3e04047397cc1c41c65c0e687 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=B4=B9=E6=94=BF=E8=81=AA?= <3300949577@qq.com>
Date: Sun, 18 Aug 2024 06:22:41 -0500
Subject: [PATCH] Update README.md

---
 README.md | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index b1a0712..e1146f5 100644
--- a/README.md
+++ b/README.md
@@ -10,9 +10,10 @@ DiT-MoE as a sparse version of the diffusion Transformer, is scalable and compet
 
 * 🪐 A PyTorch [implementation](models.py) of DiT-MoE
 * ⚡️ Pre-trained checkpoints in paper
+* 🌋 **Rectified flow**-based training and sampling scripts 
 * 💥 A [sampling script](sample.py) for running pre-trained DiT-MoE 
 * 🛸 A DiT-MoE training script using PyTorch [DDP](train.py) and [deepspeed](train_deepspeed.py)
-* 🌋 Rectified flow sampling strategy 
+
 
 ### To-do list