Hugging Face's logo

Spaces:
Culda
/
flux-inpaint-controlnet
Running

App Files Community
Culda commited on
Commit
5de3afe
•
1 Parent(s): 0c68dbd

flux schnell

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from diffusers.pipelines.flux.pipeline_flux import FluxPipeline
4
  from diffusers.models.controlnet_flux import FluxControlNetModel
5
  from controlnet_aux import CannyDetector
6
 
7
- base_model = 'black-forest-labs/FLUX.1-dev'
8
  controlnet_model = 'YishaoAI/flux-dev-controlnet-canny-kid-clothes'
9
 
10
  controlnet = FluxControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.float16)
 
4
  from diffusers.models.controlnet_flux import FluxControlNetModel
5
  from controlnet_aux import CannyDetector
6
 
7
+ base_model = 'black-forest-labs/FLUX.1-schnell'
8
  controlnet_model = 'YishaoAI/flux-dev-controlnet-canny-kid-clothes'
9
 
10
  controlnet = FluxControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.float16)