Update README.md

Including the repo link for accelerating the training process with a few new features.
facebookresearch · Jun 2, 2023 · ed81ce2 · ed81ce2
1 parent 796c29e
commit ed81ce2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/README.md b/README.md
@@ -116,6 +116,7 @@ Basic features that would be nice to add:
 - [ ] Resume training from a checkpoint
 - [ ] AMP/bfloat16 support
 
+**🔥 Feature Update** Check out this repository at https://github.com/chuanyangjin/fast-DiT to preview a selection of training speed acceleration and memory saving features including gradient checkpointing, mixed precision training and pre-extrated VAE features. With these advancements, we have achieved a training speed of 0.84 steps/sec for DiT-XL/2 using just a single A100 GPU.
 
 ## Evaluation (FID, Inception Score, etc.)