jerpint commited on
Commit
87470f9
·
1 Parent(s): 4a6a9ac

add safety checker

Browse files
Files changed (1) hide show
  1. controlnet.py +1 -1
controlnet.py CHANGED
@@ -20,7 +20,7 @@ class QRControlNet:
20
  pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
21
  "runwayml/stable-diffusion-v1-5",
22
  controlnet=controlnet,
23
- safety_checker=None,
24
  torch_dtype=torch_dtype,
25
  ).to(device)
26
 
 
20
  pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
21
  "runwayml/stable-diffusion-v1-5",
22
  controlnet=controlnet,
23
+ # safety_checker=None,
24
  torch_dtype=torch_dtype,
25
  ).to(device)
26