Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ def process_controlnet_img(image):
|
|
17 |
# load pipelines
|
18 |
pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell",
|
19 |
torch_dtype=torch.bfloat16)
|
20 |
-
pipe.enable_model_cpu_offload()
|
21 |
t5_slider = T5SliderFlux(pipe, device=torch.device("cuda"))
|
22 |
|
23 |
# pipe_adapter = StableDiffusionXLPipeline.from_pretrained("sd-community/sdxl-flash").to("cuda", torch.float16)
|
|
|
17 |
# load pipelines
|
18 |
pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell",
|
19 |
torch_dtype=torch.bfloat16)
|
20 |
+
#pipe.enable_model_cpu_offload()
|
21 |
t5_slider = T5SliderFlux(pipe, device=torch.device("cuda"))
|
22 |
|
23 |
# pipe_adapter = StableDiffusionXLPipeline.from_pretrained("sd-community/sdxl-flash").to("cuda", torch.float16)
|