Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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)
|