Hugging Face's logo

Spaces:
guardiancc
/
FLUX-LoRA-DLC
Paused

App Files Community
guardiancc commited on
Commit
e5dcc30
·
verified ·
1 Parent(s): 4201ea3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1437,9 +1437,9 @@ pipe_i2i = AutoPipelineForImage2Image.from_pretrained(base_model,
1437
 
1438
  # CONTROLNET
1439
  controlnet_model = "InstantX/FLUX.1-dev-controlnet-canny"
1440
- controlnet = FluxControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
1441
  pipe_canny = FluxControlNetPipeline.from_pretrained(
1442
- base_model, controlnet=controlnet, torch_dtype=torch.bfloat16
1443
  )
1444
 
1445
  MAX_SEED = 2**32-1
 
1437
 
1438
  # CONTROLNET
1439
  controlnet_model = "InstantX/FLUX.1-dev-controlnet-canny"
1440
+ controlnet = FluxControlNetModel.from_pretrained(controlnet_model, torch_dtype=dtype)
1441
  pipe_canny = FluxControlNetPipeline.from_pretrained(
1442
+ base_model, controlnet=controlnet, torch_dtype=dtype
1443
  )
1444
 
1445
  MAX_SEED = 2**32-1