Update README.md

README.md

@@ -18,7 +18,7 @@ Example chosen frame of GIF from CLIP
 
 ## Training Details
 
-The text encoder was trained along with the UNet at half precision for 15% of the total 8,000 steps (1,200 steps), and then the UNet was trained alone for the rest. I used a polynomial learning rate decay starting at 2e-6 (the default in fast-DreamBooth).
+Stage 1 (0-12k steps) The text encoder was trained along with the UNet at half precision for 15% of the total 8,000 steps (1,200 steps), and then the UNet was trained alone for the rest. I used a polynomial learning rate decay starting at 2e-6 (the default in fast-DreamBooth).
 
 
 ## How to use with `diffusers` library (section from [openjourney](https://huggingface.co/openjourney/openjourney))