Hugging Face's logo

Spaces:
jiuface
/
flux-controlnet-inpainting-v2
Running on Zero

App Files Community
jiuface commited on
Commit
222119d
·
verified ·
1 Parent(s): 5031a77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -38,9 +38,8 @@ IMAGE_SIZE = 1024
38
  device = "cuda" if torch.cuda.is_available() else "cpu"
39
  base_model = "black-forest-labs/FLUX.1-dev"
40
 
41
- controlnet_model = 'black-forest-labs/FLUX.1-Canny-dev'
42
- #controlnet = FluxControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
43
- controlnet = FluxControlPipeline.from_pretrained("black-forest-labs/FLUX.1-Canny-dev", torch_dtype=torch.bfloat16)
44
 
45
 
46
  pipe = FluxControlNetInpaintPipeline.from_pretrained(base_model, controlnet=controlnet, torch_dtype=torch.bfloat16).to(device)
 
38
  device = "cuda" if torch.cuda.is_available() else "cpu"
39
  base_model = "black-forest-labs/FLUX.1-dev"
40
 
41
+ controlnet_model = 'InstantX/FLUX.1-dev-Controlnet-Canny'
42
+ controlnet = FluxControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
 
43
 
44
 
45
  pipe = FluxControlNetInpaintPipeline.from_pretrained(base_model, controlnet=controlnet, torch_dtype=torch.bfloat16).to(device)