patrickvonplaten commited on
Commit
714a42b
1 Parent(s): f3a2773

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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
- "fusing/stable-diffusion-v1-5-controlnet-seg", torch_dtype=torch.float16
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(