wanghaofan
commited on
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.
|