Update README.md
Browse files
README.md
CHANGED
|
@@ -28,7 +28,7 @@ This repository contains a Depth ControlNet for FLUX.1-dev model jointly trained
|
|
| 28 |
|
| 29 |
# Model Cards
|
| 30 |
- The model consists of 4 FluxTransformerBlock and 1 FluxSingleTransformerBlock.
|
| 31 |
-
- This checkpoint is trained on both real and generated image datasets
|
| 32 |
- The recommended controlnet_conditioning_scale is 0.3-0.7.
|
| 33 |
|
| 34 |
# Showcases
|
|
@@ -74,4 +74,4 @@ For multi-ControlNets support, please refer to [Shakker-Labs/FLUX.1-dev-ControlN
|
|
| 74 |
- [Shakker-Labs/FLUX.1-dev-ControlNet-Union-Pro](https://huggingface.co/Shakker-Labs/FLUX.1-dev-ControlNet-Union-Pro)
|
| 75 |
|
| 76 |
# Acknowledgements
|
| 77 |
-
This project is released by [Shakker AI](https://www.shakker.ai/). All copyright reserved.
|
|
|
|
| 28 |
|
| 29 |
# Model Cards
|
| 30 |
- The model consists of 4 FluxTransformerBlock and 1 FluxSingleTransformerBlock.
|
| 31 |
+
- This checkpoint is trained on both real and generated image datasets, with 16\*A800 for 70K steps. The batch size 16\*4=64 with resolution=1024. The learning rate is set to 5e-6. We use [Depth-Anything-V2](https://github.com/DepthAnything/Depth-Anything-V2) to extract depth maps.
|
| 32 |
- The recommended controlnet_conditioning_scale is 0.3-0.7.
|
| 33 |
|
| 34 |
# Showcases
|
|
|
|
| 74 |
- [Shakker-Labs/FLUX.1-dev-ControlNet-Union-Pro](https://huggingface.co/Shakker-Labs/FLUX.1-dev-ControlNet-Union-Pro)
|
| 75 |
|
| 76 |
# Acknowledgements
|
| 77 |
+
This project is sponsored and released by [Shakker AI](https://www.shakker.ai/). All copyright reserved.
|