Spaces:

AP123
/

IllusionDiffusion

Running on Zero

AP123 commited on May 14

Commit

ee36d88

•

1 Parent(s): 766763f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

     BASE_MODEL,
     controlnet=controlnet,
     vae=vae,
+    safety_checker=safety_checker,
     torch_dtype=torch.float16,
 ).to("cuda")