linoyts HF staff commited on
Commit
3753c1a
1 Parent(s): efe76f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -26,9 +26,9 @@ t5_slider = T5SliderFlux(pipe, device=torch.device("cuda"))
26
 
27
  base_model = 'black-forest-labs/FLUX.1-schnell'
28
  controlnet_model = 'InstantX/FLUX.1-dev-Controlnet-Canny-alpha'
29
- controlnet = FluxControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
30
- pipe_controlnet = FluxControlNetPipeline.from_pretrained(base_model, controlnet=controlnet, torch_dtype=torch.bfloat16)
31
- t5_slider_controlnet = T5SliderFlux(sd_pipe=pipe_controlnet,device=torch.device("cuda"))
32
 
33
 
34
  @spaces.GPU(duration=120)
 
26
 
27
  base_model = 'black-forest-labs/FLUX.1-schnell'
28
  controlnet_model = 'InstantX/FLUX.1-dev-Controlnet-Canny-alpha'
29
+ # controlnet = FluxControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
30
+ # pipe_controlnet = FluxControlNetPipeline.from_pretrained(base_model, controlnet=controlnet, torch_dtype=torch.bfloat16)
31
+ # t5_slider_controlnet = T5SliderFlux(sd_pipe=pipe_controlnet,device=torch.device("cuda"))
32
 
33
 
34
  @spaces.GPU(duration=120)