Hugging Face's logo

Spaces:
Shakker-Labs
/
FLUX.1-dev-ControlNet-Union-Pro
Running on Zero

App Files Community
2
wanghaofan commited on
Commit
c30ca1a
•
1 Parent(s): dfd025a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -39,7 +39,8 @@ model = model.to(DEVICE).eval()
39
 
40
  import torch
41
  from diffusers.utils import load_image
42
- from diffusers import FluxControlNetPipeline, FluxControlNetModel, FluxMultiControlNetModel
 
43
 
44
  base_model = 'black-forest-labs/FLUX.1-dev'
45
  controlnet_model = 'Shakker-Labs/FLUX.1-dev-ControlNet-Union-Pro'
 
39
 
40
  import torch
41
  from diffusers.utils import load_image
42
+ from diffusers import FluxControlNetPipeline, FluxControlNetModel
43
+ from diffusers.models import FluxMultiControlNetModel
44
 
45
  base_model = 'black-forest-labs/FLUX.1-dev'
46
  controlnet_model = 'Shakker-Labs/FLUX.1-dev-ControlNet-Union-Pro'