latentnavigation-flux

Runtime error

linoyts HF staff commited on Aug 21

Commit

058c742

•

1 Parent(s): 325d5c6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ def process_controlnet_img(image):
     controlnet_img = Image.fromarray(controlnet_img)
 # load pipelines
-pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev",
                                     torch_dtype=torch.bfloat16)
 pipe.enable_model_cpu_offload()
 t5_slider = T5SliderFlux(pipe, device=torch.device("cuda"))

     controlnet_img = Image.fromarray(controlnet_img)
 # load pipelines
+pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell",
                                     torch_dtype=torch.bfloat16)
 pipe.enable_model_cpu_offload()
 t5_slider = T5SliderFlux(pipe, device=torch.device("cuda"))