Update README.md
Browse files
README.md
CHANGED
@@ -9,17 +9,20 @@ inference: false
|
|
9 |
|
10 |
---
|
11 |
|
12 |
-
**The models in this respository are a preview for the final model. Training is still in progress.**
|
13 |
-
|
14 |
# Waifu Diffusion v1.3
|
15 |
|
16 |
Waifu Diffusion is a latent text-to-image diffusion model that has been conditioned on high-quality anime images through fine-tuning.
|
17 |
|
|
|
|
|
|
|
|
|
|
|
18 |
## Model Description
|
19 |
|
20 |
-
The model originally used for fine-tuning is [Stable Diffusion
|
21 |
|
22 |
-
|
23 |
|
24 |
## License
|
25 |
|
|
|
9 |
|
10 |
---
|
11 |
|
|
|
|
|
12 |
# Waifu Diffusion v1.3
|
13 |
|
14 |
Waifu Diffusion is a latent text-to-image diffusion model that has been conditioned on high-quality anime images through fine-tuning.
|
15 |
|
16 |
+
- [Float 16 EMA Pruned](https://huggingface.co/hakurei/waifu-diffusion-v1-3/blob/main/wd-v1-3-float16.ckpt)
|
17 |
+
- [Float 32 EMA Pruned](https://huggingface.co/hakurei/waifu-diffusion-v1-3/blob/main/wd-v1-3-float32.ckpt)
|
18 |
+
- [Float 32 Full Weights](https://huggingface.co/hakurei/waifu-diffusion-v1-3/blob/main/wd-v1-3-full.ckpt)
|
19 |
+
- [Float 32 Full Weights + Optimizer Weights (For Training)](https://huggingface.co/hakurei/waifu-diffusion-v1-3/blob/main/wd-v1-3-full-opt.ckpt)
|
20 |
+
|
21 |
## Model Description
|
22 |
|
23 |
+
The model originally used for fine-tuning is [Stable Diffusion 1.4](https://huggingface.co/CompVis/stable-diffusion-v1-4), which is a latent image diffusion model trained on [LAION2B-en](https://huggingface.co/datasets/laion/laion2B-en). The current model has been fine-tuned with a learning rate of 5.0e-6 for 10 epochs on 680k anime-styled images.
|
24 |
|
25 |
+
[See here for an in-depth overview of Waifu Diffusion 1.3.](https://gist.github.com/harubaru/f727cedacae336d1f7877c4bbe2196e1)
|
26 |
|
27 |
## License
|
28 |
|