wanghaofan commited on
Commit
482abfe
·
verified ·
1 Parent(s): b443b30

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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. with 16\*A800 for 50K steps. The batch size 16\*4=64 with resolution=1024. The learning rate is set to 5e-6.
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.