wanghaofan commited on
Commit
46371a6
1 Parent(s): a321e4c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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 = './Shakker-Labs/FLUX.1-dev-Controlnet-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,7 +88,7 @@ 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'
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'
 
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'