Commit
·
714a42b
1
Parent(s):
f3a2773
Update README.md
Browse files
README.md
CHANGED
@@ -111,7 +111,7 @@ color_seg = color_seg.astype(np.uint8)
|
|
111 |
image = Image.fromarray(color_seg)
|
112 |
|
113 |
controlnet = ControlNetModel.from_pretrained(
|
114 |
-
"
|
115 |
)
|
116 |
|
117 |
pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
|
|
111 |
image = Image.fromarray(color_seg)
|
112 |
|
113 |
controlnet = ControlNetModel.from_pretrained(
|
114 |
+
"lllyasviel/sd-controlnet-seg", torch_dtype=torch.float16
|
115 |
)
|
116 |
|
117 |
pipe = StableDiffusionControlNetPipeline.from_pretrained(
|