Upload example images for hitokomoru-15000.ckpt and hitokomoru-20000.ckpt

README.md

@@ -24,25 +24,63 @@ There is 4 variations of this model available so far:
 # Dataset
 
 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)
- 
+
+## 🧨 Diffusers
+
+This model can be used just like any other Stable Diffusion model. For more information,
+please have a look at the [Stable Diffusion](https://huggingface.co/docs/diffusers/api/pipelines/stable_diffusion).
+
+You can also export the model to [ONNX](https://huggingface.co/docs/diffusers/optimization/onnx), [MPS](https://huggingface.co/docs/diffusers/optimization/mps) and/or [FLAX/JAX]().
+
+```python
+from diffusers import StableDiffusionPipeline
+import torch
+
+model_id = "Linaqruf/hitokomoru-diffusion"
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+pipe = pipe.to("cuda")
+
+prompt = "hatsune_miku
+image = pipe(prompt).images[0]
+
+image.save("./hatsune_miku.png")
+```
+
 ## Examples
 
 Below are some examples of images generated using this model:
 
-#### Using Hitokomoru-5000.ckpt
+#### Using Hitokomoru-5000-pruned.ckpt
 **Anime Girl:**
-![Anime Girl](https://huggingface.co/Linaqruf/hitokomoru/resolve/main/girl-5000.png)
+![Anime Girl](https://huggingface.co/Linaqruf/hitokomoru/resolve/main/example_images/girl-5000.png)
 
 **Anime Boy:**
-![Anime Boy](https://huggingface.co/Linaqruf/hitokomoru/resolve/main/boy-5000.png)
+![Anime Boy](https://huggingface.co/Linaqruf/hitokomoru/resolve/main/example_images/boy-5000.png)
 
-#### Using Hitokomoru-10000.ckpt
+#### Using Hitokomoru-10000-pruned.ckpt
 
 **Anime Girl:**
-![Anime Girl](https://huggingface.co/Linaqruf/hitokomoru/resolve/main/girl-10000.png)
+![Anime Girl](https://huggingface.co/Linaqruf/hitokomoru/resolve/main/example_images/girl-10000.png)
 
 **Anime Boy:**
-![Anime Boy](https://huggingface.co/Linaqruf/hitokomoru/resolve/main/boy-10000.png)
+![Anime Boy](https://huggingface.co/Linaqruf/hitokomoru/resolve/main/example_images/boy-10000.png)
+
+#### Using Hitokomoru-15000-pruned.ckpt
+
+**Anime Girl:**
+![Anime Girl](https://huggingface.co/Linaqruf/hitokomoru/resolve/main/example_images/girl-15000.png)
+
+**Anime Boy:**
+![Anime Boy](https://huggingface.co/Linaqruf/hitokomoru/resolve/main/example_images/boy-15000.png)
+
+#### Using Hitokomoru-20000-pruned.ckpt
+
+**Anime Girl:**
+![Anime Girl](https://huggingface.co/Linaqruf/hitokomoru/resolve/main/example_images/girl-20000.png)
+
+**Anime Boy:**
+![Anime Boy](https://huggingface.co/Linaqruf/hitokomoru/resolve/main/example_images/boy-20000.png)
+
 
 ## License