patrickvonplaten
commited on
Commit
•
a730f6c
1
Parent(s):
1e0b21f
Update README.md
Browse files
README.md
CHANGED
@@ -101,7 +101,7 @@ image = load_image("https://huggingface.co/lllyasviel/sd-controlnet-hed/resolve/
|
|
101 |
image = hed(image)
|
102 |
|
103 |
controlnet = ControlNetModel.from_pretrained(
|
104 |
-
"
|
105 |
)
|
106 |
|
107 |
pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
|
|
101 |
image = hed(image)
|
102 |
|
103 |
controlnet = ControlNetModel.from_pretrained(
|
104 |
+
"lllyasviel/sd-controlnet-hed", torch_dtype=torch.float16
|
105 |
)
|
106 |
|
107 |
pipe = StableDiffusionControlNetPipeline.from_pretrained(
|