Update README.md

README.md

@@ -4,7 +4,7 @@ license: creativeml-openrail-m
 
 # 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. The model is trained using the [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 10000 training steps on 255 Hitokomoru artwork. 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_** 
 
@@ -16,14 +16,14 @@ You can find datasets used to train this model [here](https://huggingface.co/dat
 
 Below are some examples of images generated using this model:
 
-### Using Hitomoru-5000.ckpt
+#### Using Hitokomoru-5000.ckpt
 **Anime Girl:**
 ![Anime Girl](https://huggingface.co/Linaqruf/hitokomoru/resolve/main/girl-5000.png)
 
 **Anime Boy:**
 ![Anime Boy](https://huggingface.co/Linaqruf/hitokomoru/resolve/main/boy-5000.png)
 
-### Using Hitomoru-10000.ckpt
+#### Using Hitokomoru-10000.ckpt
 
 **Anime Girl:**
 ![Anime Girl](https://huggingface.co/Linaqruf/hitokomoru/resolve/main/girl-10000.png)
@@ -41,8 +41,13 @@ The CreativeML OpenRAIL License specifies:
 3. You may re-distribute the weights and use the model commercially and/or as a service. If you do, please be aware you have to include the same use restrictions as the ones in the license and share a copy of the CreativeML OpenRAIL-M to all your users (please read the license entirely and carefully)
 [Please read the full license here](https://huggingface.co/spaces/CompVis/stable-diffusion-license)
 
+## What's next?
+- Trying to resuming training until 20k or even 50k if possible (Research purpose, if the result is great I'll upload here)
+- Adding example images for `Scenery`
+
 ## Credit
 - [ヒトこもる/Hitokomoru](https://www.pixiv.net/en/users/30837811) for Datasets
 
 ## Big Thanks to
-- [Kohya](https://twitter.com/kohya_ss) with their [Kohya Trainer](https://note.com/kohya_ss/n/ne17e34dd51bf)
+- [Kohya](https://twitter.com/kohya_ss) with their [Kohya Trainer](https://note.com/kohya_ss/n/ne17e34dd51bf)
+- Peeps on SD Training Labs Discord Server