jasperai
/

Flux.1-dev-Controlnet-Depth

Model card Files Files and versions

clementchadebec commited on Sep 24, 2024

Commit

adc1f0e

·

verified ·

1 Parent(s): 8244a2e

Update README.md

Files changed (1) hide show

README.md +2 -0

README.md CHANGED Viewed

@@ -69,6 +69,8 @@ from diffusers.utils import load_image
 midas = MidasDetector.from_pretrained("lllyasviel/Annotators")
 # Load an image
 im = load_image(
   "https://huggingface.co/jasperai/jasperai/Flux.1-dev-Controlnet-Depth/resolve/main/examples/output.jpg"

 midas = MidasDetector.from_pretrained("lllyasviel/Annotators")
+midas.to("cuda")
 # Load an image
 im = load_image(
   "https://huggingface.co/jasperai/jasperai/Flux.1-dev-Controlnet-Depth/resolve/main/examples/output.jpg"