Spaces:
Running
Running
add safety checker
Browse files- 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 |
|