mishig HF staff commited on
Commit
1e8371c
1 Parent(s): 8e580f2

Use controlnet v1.1

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ async (canvas, prompt) => {
39
 
40
  # Models
41
  controlnet = ControlNetModel.from_pretrained(
42
- "lllyasviel/sd-controlnet-canny", torch_dtype=torch.float16
43
  )
44
  pipe = StableDiffusionControlNetPipeline.from_pretrained(
45
  "runwayml/stable-diffusion-v1-5", controlnet=controlnet, safety_checker=None, torch_dtype=torch.float16
 
39
 
40
  # Models
41
  controlnet = ControlNetModel.from_pretrained(
42
+ "lllyasviel/control_v11p_sd15_canny", torch_dtype=torch.float16
43
  )
44
  pipe = StableDiffusionControlNetPipeline.from_pretrained(
45
  "runwayml/stable-diffusion-v1-5", controlnet=controlnet, safety_checker=None, torch_dtype=torch.float16