Hugging Face's logo

Shakker-Labs
/
FLUX.1-dev-ControlNet-Union-Pro

Text-to-Image
Diffusers
Safetensors
English
Text-to-Image
ControlNet
Diffusers
Flux.1-dev
image-generation
Stable Diffusion
Model card Files
xet
 Community
31
wanghaofan commited on
Commit
66c6bd0
·
verified ·
1 Parent(s): a320fd8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -2
README.md CHANGED
@@ -51,7 +51,8 @@ Please install `diffusers` from [the source](https://github.com/huggingface/diff
51
  import torch
52
  from diffusers.utils import load_image
53
 
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'
@@ -86,7 +87,8 @@ image = pipe(
86
 
87
  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 = 'Shakker-Labs/FLUX.1-dev-ControlNet-Union-Pro'
92
  controlnet_union = FluxControlNetModel.from_pretrained(controlnet_model_union, torch_dtype=torch.bfloat16)
 
51
  import torch
52
  from diffusers.utils import load_image
53
 
54
+ from diffusers import FluxControlNetPipeline, FluxControlNetModel
55
+ from diffusers.models import FluxMultiControlNetModel
56
 
57
  base_model = 'black-forest-labs/FLUX.1-dev'
58
  controlnet_model_union = 'Shakker-Labs/FLUX.1-dev-ControlNet-Union-Pro'
 
87
 
88
  We also support loading multiple ControlNets as before, you can load as
89
  ```python
90
+ from diffusers import FluxControlNetModel
91
+ from diffusers.models import FluxMultiControlNetModel
92
 
93
  controlnet_model_union = 'Shakker-Labs/FLUX.1-dev-ControlNet-Union-Pro'
94
  controlnet_union = FluxControlNetModel.from_pretrained(controlnet_model_union, torch_dtype=torch.bfloat16)