Update README.md

2026-07-26 22:30:28 +00:00 · 2024-08-18 06:22:41 -05:00
parent 547a0961c8
commit 8c16f12fb4
1 changed files with 2 additions and 1 deletions
@@ -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