Update README.md

README.md

@@ -5,13 +5,19 @@ thumbnail: "https://huggingface.co/Linaqruf/hitokomoru/resolve/main/thumbnail.pn
 
 # Hitokomoru Diffusion
 
-**Welcome to Hitokomoru Diffusion** - a latent diffusion model that has been trained on Japanese Artist artwork, [ヒトこもる/Hitokomoru](https://www.pixiv.net/en/users/30837811). The current model has been fine-tuned with a learning rate of `2.0e-6` for `10000 training steps` on `255 images` collected from Danbooru. The model is trained using [NovelAI Aspect Ratio Bucketing Tool](https://github.com/NovelAI/novelai-aspect-ratio-bucketing) so that it can be trained at non-square resolutions. Like other anime-style Stable Diffusion models, it also supports Danbooru tags to generate images.
+**Welcome to Hitokomoru Diffusion** - a latent diffusion model that has been trained on Japanese Artist artwork, [ヒトこもる/Hitokomoru](https://www.pixiv.net/en/users/30837811). The current model has been fine-tuned with a learning rate of `2.0e-6` for `20000 training steps`/`80 Epochs` on `255 images` collected from Danbooru. The model is trained using [NovelAI Aspect Ratio Bucketing Tool](https://github.com/NovelAI/novelai-aspect-ratio-bucketing) so that it can be trained at non-square resolutions. Like other anime-style Stable Diffusion models, it also supports Danbooru tags to generate images.
 
 e.g. **_1girl, white hair, golden eyes, beautiful eyes, detail, flower meadow, cumulonimbus clouds, lighting, detailed sky, garden_** 
 
+There is 4 variations of this model available so far:
+- `hitokomoru-5000.ckpt for the checkpoint trained on 5,000 steps.`
+- `hitokomoru-10000.ckpt for the checkpoint trained on 10,000 steps.`
+- `hitokomoru-15000.ckpt for the checkpoint trained on 15,000 steps.`
+- `hitokomoru-20000.ckpt for the checkpoint trained on 20,000 steps.`
+
 # Dataset
 
-You can find datasets used to train this model [here](https://huggingface.co/datasets/Linaqruf/hitokomoru-tag)
+You can find `datasets` used to train this model and the `last-state` folder for resume training [here](https://huggingface.co/datasets/Linaqruf/hitokomoru-tag)
  
 ## Examples