Hugging Face's logo

Spaces:
Culda
/
flux-inpaint-controlnet
Running

App Files Community
Culda commited on
Commit
5a82940
•
1 Parent(s): e42f58a

no auto

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -22,11 +22,9 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
22
  base_model = "black-forest-labs/FLUX.1-dev"
23
  controlnet_model = "YishaoAI/flux-dev-controlnet-canny-kid-clothes"
24
 
25
- controlnet = FluxControlNetModel.from_pretrained(
26
- controlnet_model, torch_dtype=dtype
27
- )
28
  pipe = FluxControlNetInpaintPipeline.from_pretrained(
29
- base_model, controlnet=controlnet, torch_dtype=dtype, device_map="auto"
30
  ).to(device)
31
 
32
  pipe.enable_model_cpu_offload()
 
22
  base_model = "black-forest-labs/FLUX.1-dev"
23
  controlnet_model = "YishaoAI/flux-dev-controlnet-canny-kid-clothes"
24
 
25
+ controlnet = FluxControlNetModel.from_pretrained(controlnet_model, torch_dtype=dtype)
 
 
26
  pipe = FluxControlNetInpaintPipeline.from_pretrained(
27
+ base_model, controlnet=controlnet, torch_dtype=dtype
28
  ).to(device)
29
 
30
  pipe.enable_model_cpu_offload()