Update README.md
Browse files
README.md
CHANGED
|
@@ -54,7 +54,7 @@ from diffusers.utils import load_image
|
|
| 54 |
from diffusers import FluxControlNetPipeline, FluxControlNetModel, FluxMultiControlNetModel
|
| 55 |
|
| 56 |
base_model = 'black-forest-labs/FLUX.1-dev'
|
| 57 |
-
controlnet_model_union = '
|
| 58 |
|
| 59 |
controlnet_union = FluxControlNetModel.from_pretrained(controlnet_model_union, torch_dtype=torch.bfloat16)
|
| 60 |
controlnet = FluxMultiControlNetModel([controlnet_union]) # we always recommend loading via FluxMultiControlNetModel
|
|
@@ -88,10 +88,10 @@ We also support loading multiple ControlNets as before, you can load as
|
|
| 88 |
```python
|
| 89 |
from diffusers import FluxControlNetModel, FluxMultiControlNetModel
|
| 90 |
|
| 91 |
-
controlnet_model_union = '
|
| 92 |
controlnet_union = FluxControlNetModel.from_pretrained(controlnet_model_union, torch_dtype=torch.bfloat16)
|
| 93 |
|
| 94 |
-
controlnet_model_depth = '
|
| 95 |
controlnet_depth = FluxControlNetModel.from_pretrained(controlnet_model_depth, torch_dtype=torch.bfloat16)
|
| 96 |
|
| 97 |
controlnet = FluxMultiControlNetModel([controlnet_union, controlnet_depth])
|
|
|
|
| 54 |
from diffusers import FluxControlNetPipeline, FluxControlNetModel, FluxMultiControlNetModel
|
| 55 |
|
| 56 |
base_model = 'black-forest-labs/FLUX.1-dev'
|
| 57 |
+
controlnet_model_union = 'Shakker-Labs/FLUX.1-dev-ControlNet-Union-Pro'
|
| 58 |
|
| 59 |
controlnet_union = FluxControlNetModel.from_pretrained(controlnet_model_union, torch_dtype=torch.bfloat16)
|
| 60 |
controlnet = FluxMultiControlNetModel([controlnet_union]) # we always recommend loading via FluxMultiControlNetModel
|
|
|
|
| 88 |
```python
|
| 89 |
from diffusers import FluxControlNetModel, FluxMultiControlNetModel
|
| 90 |
|
| 91 |
+
controlnet_model_union = 'Shakker-Labs/FLUX.1-dev-ControlNet-Union-Pro'
|
| 92 |
controlnet_union = FluxControlNetModel.from_pretrained(controlnet_model_union, torch_dtype=torch.bfloat16)
|
| 93 |
|
| 94 |
+
controlnet_model_depth = 'Shakker-Labs/FLUX.1-dev-Controlnet-Depth'
|
| 95 |
controlnet_depth = FluxControlNetModel.from_pretrained(controlnet_model_depth, torch_dtype=torch.bfloat16)
|
| 96 |
|
| 97 |
controlnet = FluxMultiControlNetModel([controlnet_union, controlnet_depth])
|