linoyts HF staff commited on
Commit
f4548f6
1 Parent(s): 226559a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)