Update README.md

README.md

@@ -11,10 +11,13 @@ The idea of the "warmup stages" is to sort of pretrain the LoRA weights so that
 
 ***This stage is 256p, which is what Qwen-Image originally pretrained with.***
 
+## Training Stages
 - ***[SynthSat-v1.0-warmup-stage1](https://huggingface.co/PJMixers-Images/Qwen-Image-SynthSat-v1.0-warmup-stage1) is 256p only, using freshly initialized weights.***
 - [SynthSat-v1.0-warmup-stage2](https://huggingface.co/PJMixers-Images/Qwen-Image-SynthSat-v1.0-warmup-stage2) is 640p only, re-using the weights from [SynthSat-v1.0-warmup-stage1](https://huggingface.co/PJMixers-Images/Qwen-Image-SynthSat-v1.0-warmup-stage1).
-- SynthSat-v1.0-warmup-stage3 is 640p+1328p, re-using the weights from [SynthSat-v1.0-warmup-stage2](https://huggingface.co/PJMixers-Images/Qwen-Image-SynthSat-v1.0-warmup-stage2).
-- SynthSat-v1.0 is 640p+1328p+1536p, re-using the weights from SynthSat-v1.0-warmup-stage3.
+- [SynthSat-v1.0-warmup-stage2.5-nsfw](https://huggingface.co/PJMixers-Images/Qwen-Image-SynthSat-v1.0-warmup-stage2.5-nsfw) is 256p+640p, re-using the weights from [SynthSat-v1.0-warmup-stage2-step1400](https://huggingface.co/PJMixers-Images/Qwen-Image-SynthSat-v1.0-warmup-stage2/blob/main/Qwen-Image-SynthSat-v1.0-warmup-stage2-step00001400.safetensors), using only NSFW images.
+- Next stage is either warmup-stage3 doing 640p+1328p or v1.0 doing 640p+1328p+1536p on a more refined dataset.
+
+## Current Stage Details
 
 ```
 INFO:musubi_tuner.hv_train_network:Load dataset config from /media/xzuyn/Toshiba1/musubi-stuff/dataset_configs/SynthSat-v1.0-warmup-stage1.toml