jasperai
/

Flux.1-dev-Controlnet-Depth

Model card Files Files and versions Community

clementchadebec commited on Sep 23

Commit

70f8801

•

1 Parent(s): d19b7e5

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -36,7 +36,7 @@ pipe = FluxControlNetPipeline.from_pretrained(
   controlnet=controlnet,
   torch_dtype=torch.bfloat16
 )
 # Load a control image
 control_image = load_image(

   controlnet=controlnet,
   torch_dtype=torch.bfloat16
 )
+pipe.to("cuda")
 # Load a control image
 control_image = load_image(