Spaces:

AP123
/

IllusionDiffusion

Running on Zero

AP123 commited on May 14

Commit

a2f5ebe

•

1 Parent(s): 884858d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -31,7 +31,7 @@ main_pipe = StableDiffusionControlNetPipeline.from_pretrained(
     BASE_MODEL,
     controlnet=controlnet,
     vae=vae,
-    safety_checker=None,
     torch_dtype=torch.float16,
 ).to("cuda")

     BASE_MODEL,
     controlnet=controlnet,
     vae=vae,
+    safety_checker=StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker"),
     torch_dtype=torch.float16,
 ).to("cuda")