hakurei commited on
Commit
90a6846
1 Parent(s): 5b0dd1c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -4
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 V1-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).
21
 
22
- The current model has been fine-tuned with a learning rate of 5.0e-6 for 10 epochs on 680k anime-styled images.
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