Update README.md
Browse files
README.md
CHANGED
@@ -9,6 +9,8 @@ inference: true
|
|
9 |
|
10 |
---
|
11 |
|
|
|
|
|
12 |
This unofficial repository hosts a diffusers-compatible float16 checkpoint of the [WDXL](https://huggingface.co/hakurei/waifu-diffusion-xl) base UNet.
|
13 |
|
14 |
For convenience (i.e. for use in a StableDiffusionXLPipeline) we include mirrors of other models (please adhere to their terms of usage):
|
@@ -38,7 +40,6 @@ from typing import List
|
|
38 |
scheduler: DPMSolverMultistepScheduler = DPMSolverMultistepScheduler.from_pretrained(
|
39 |
'Birchlabs/waifu-diffusion-xl-unofficial',
|
40 |
subfolder='scheduler',
|
41 |
-
# sde-dpmsolver++ is very new. if your diffusers version doesn't have it: use 'dpmsolver++' instead.
|
42 |
algorithm_type='sde-dpmsolver++',
|
43 |
solver_order=2,
|
44 |
# solver_type='heun' may give a sharper image. Cheng Lu reckons midpoint is better.
|
@@ -81,6 +82,10 @@ img, *_ = images
|
|
81 |
img.save('waifu.png')
|
82 |
```
|
83 |
|
|
|
|
|
|
|
|
|
84 |
### UNet2DConditionModel
|
85 |
|
86 |
If you just want the UNet, you can load it like so:
|
|
|
9 |
|
10 |
---
|
11 |
|
12 |
+
|
13 |
+
|
14 |
This unofficial repository hosts a diffusers-compatible float16 checkpoint of the [WDXL](https://huggingface.co/hakurei/waifu-diffusion-xl) base UNet.
|
15 |
|
16 |
For convenience (i.e. for use in a StableDiffusionXLPipeline) we include mirrors of other models (please adhere to their terms of usage):
|
|
|
40 |
scheduler: DPMSolverMultistepScheduler = DPMSolverMultistepScheduler.from_pretrained(
|
41 |
'Birchlabs/waifu-diffusion-xl-unofficial',
|
42 |
subfolder='scheduler',
|
|
|
43 |
algorithm_type='sde-dpmsolver++',
|
44 |
solver_order=2,
|
45 |
# solver_type='heun' may give a sharper image. Cheng Lu reckons midpoint is better.
|
|
|
82 |
img.save('waifu.png')
|
83 |
```
|
84 |
|
85 |
+
You should get a picture like this:
|
86 |
+
|
87 |
+
<img width="384px" height="384px" src="https://birchlabs.co.uk/share/wdxl-unofficial/0_48_waifu.png" title="seed 48: girl with green hair and sweater at night">
|
88 |
+
|
89 |
### UNet2DConditionModel
|
90 |
|
91 |
If you just want the UNet, you can load it like so:
|