clementchadebec commited on
Commit
adc1f0e
1 Parent(s): 8244a2e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -0
README.md CHANGED
@@ -69,6 +69,8 @@ from diffusers.utils import load_image
69
 
70
  midas = MidasDetector.from_pretrained("lllyasviel/Annotators")
71
 
 
 
72
  # Load an image
73
  im = load_image(
74
  "https://huggingface.co/jasperai/jasperai/Flux.1-dev-Controlnet-Depth/resolve/main/examples/output.jpg"
 
69
 
70
  midas = MidasDetector.from_pretrained("lllyasviel/Annotators")
71
 
72
+ midas.to("cuda")
73
+
74
  # Load an image
75
  im = load_image(
76
  "https://huggingface.co/jasperai/jasperai/Flux.1-dev-Controlnet-Depth/resolve/main/examples/output.jpg"