Text-to-Image
Diffusers
Safetensors
English
Text-to-Image
ControlNet
Diffusers
Flux.1-dev
image-generation
Stable Diffusion
Instructions to use Shakker-Labs/FLUX.1-dev-ControlNet-Depth with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use Shakker-Labs/FLUX.1-dev-ControlNet-Depth with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("Shakker-Labs/FLUX.1-dev-ControlNet-Depth", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
- Local Apps
- Draw Things
- DiffusionBee
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.
|