wanghaofan
commited on
Commit
•
c4f61d2
1
Parent(s):
46371a6
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])
|