clementchadebec commited on
Commit
1bd437a
1 Parent(s): 4ea0c76

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
  normal_bae = NormalBaeDetector.from_pretrained("lllyasviel/Annotators")
71
 
 
 
72
  # Load an image
73
  im = load_image(
74
  "https://huggingface.co/jasperai/Flux.1-dev-Controlnet-Surface-Normals/resolve/main/examples/output.jpg"
 
69
 
70
  normal_bae = NormalBaeDetector.from_pretrained("lllyasviel/Annotators")
71
 
72
+ normal_bae.to("cuda")
73
+
74
  # Load an image
75
  im = load_image(
76
  "https://huggingface.co/jasperai/Flux.1-dev-Controlnet-Surface-Normals/resolve/main/examples/output.jpg"