Update README.md
Browse files
README.md
CHANGED
@@ -1,3 +1,37 @@
|
|
1 |
---
|
2 |
license: openrail++
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
---
|
2 |
license: openrail++
|
3 |
+
base_model: stabilityai/stable-diffusion-xl-base-1.0
|
4 |
+
tags:
|
5 |
+
- stable-diffusion-xl
|
6 |
+
- stable-diffusion-xl-diffusers
|
7 |
+
- text-to-image
|
8 |
+
- diffusers
|
9 |
+
- controlnet
|
10 |
+
inference: false
|
11 |
---
|
12 |
+
|
13 |
+
# SDXL-controlnet: OpenPose (v2)
|
14 |
+
|
15 |
+
These are controlnet weights trained on stabilityai/stable-diffusion-xl-base-1.0 with OpenPose (v2) conditioning. You can find some example images in the following.
|
16 |
+
|
17 |
+
prompt: a ballerina, romantic sunset, 4k photo
|
18 |
+
![images_0)](./out_ballerina.png)
|
19 |
+
|
20 |
+
### Training
|
21 |
+
|
22 |
+
Use of the training script by HF🤗 [here](https://github.com/huggingface/diffusers/blob/main/examples/controlnet/README_sdxl.md).
|
23 |
+
|
24 |
+
#### Training data
|
25 |
+
This checkpoint was first trained for 15,000 steps on laion 6a resized to a max minimum dimension of 768.
|
26 |
+
|
27 |
+
#### Compute
|
28 |
+
one 1xA100 machine (Thanks a lot HF🤗 to provide the compute!)
|
29 |
+
|
30 |
+
#### Batch size
|
31 |
+
Data parallel with a single gpu batch size of 2 with gradient accumulation 8.
|
32 |
+
|
33 |
+
#### Hyper Parameters
|
34 |
+
Constant learning rate of 8e-5
|
35 |
+
|
36 |
+
#### Mixed precision
|
37 |
+
fp16
|