Update README.md
Browse files
README.md
CHANGED
|
@@ -109,3 +109,14 @@ gen_images = pipe(
|
|
| 109 |
gen_images.save('out_sketch.png')
|
| 110 |
```
|
| 111 |
<a href="https://huggingface.co/Adapter/t2iadapter/resolve/main/figs_SDXLV1.0/cond_sketch.png"><img width="480" style="margin:0;padding:0;" src="https://huggingface.co/Adapter/t2iadapter/resolve/main/figs_SDXLV1.0/res_sketch.png"/></a>
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 109 |
gen_images.save('out_sketch.png')
|
| 110 |
```
|
| 111 |
<a href="https://huggingface.co/Adapter/t2iadapter/resolve/main/figs_SDXLV1.0/cond_sketch.png"><img width="480" style="margin:0;padding:0;" src="https://huggingface.co/Adapter/t2iadapter/resolve/main/figs_SDXLV1.0/res_sketch.png"/></a>
|
| 112 |
+
|
| 113 |
+
### Training
|
| 114 |
+
|
| 115 |
+
Our training script was built on top of the official training script that we provide [here](https://github.com/huggingface/diffusers/blob/main/examples/t2i_adapter/README_sdxl.md).
|
| 116 |
+
|
| 117 |
+
The model is trained on 3M high-resolution image-text pairs from LAION-Aesthetics V2 with
|
| 118 |
+
|
| 119 |
+
- Training steps: 20000
|
| 120 |
+
- Batch size: Data parallel with a single gpu batch size of `16` for a total batch size of `256`.
|
| 121 |
+
- Learning rate: Constant learning rate of `1e-5`.
|
| 122 |
+
- Mixed precision: fp16
|