Update README.md

README.md

@@ -51,7 +51,8 @@ Please install `diffusers` from [the source](https://github.com/huggingface/diff
 import torch
 from diffusers.utils import load_image
 
-from diffusers import FluxControlNetPipeline, FluxControlNetModel, FluxMultiControlNetModel
+from diffusers import FluxControlNetPipeline, FluxControlNetModel
+from diffusers.models import FluxMultiControlNetModel
 
 base_model = 'black-forest-labs/FLUX.1-dev'
 controlnet_model_union = 'Shakker-Labs/FLUX.1-dev-ControlNet-Union-Pro'
@@ -86,7 +87,8 @@ image = pipe(
 
 We also support loading multiple ControlNets as before, you can load as
 ```python
-from diffusers import FluxControlNetModel, FluxMultiControlNetModel
+from diffusers import FluxControlNetModel
+from diffusers.models import FluxMultiControlNetModel
 
 controlnet_model_union = 'Shakker-Labs/FLUX.1-dev-ControlNet-Union-Pro'
 controlnet_union = FluxControlNetModel.from_pretrained(controlnet_model_union, torch_dtype=torch.bfloat16)