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 |
+

|
| 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
|