jasperai
/

Flux.1-dev-Controlnet-Surface-Normals

Model card Files Files and versions

clementchadebec commited on Sep 24, 2024

Commit

1bd437a

·

verified ·

1 Parent(s): 4ea0c76

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
 normal_bae = NormalBaeDetector.from_pretrained("lllyasviel/Annotators")
 # Load an image
 im = load_image(
   "https://huggingface.co/jasperai/Flux.1-dev-Controlnet-Surface-Normals/resolve/main/examples/output.jpg"

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