Code-Cooker

Sleeping

Severian commited on Sep 6, 2024

Commit

2060e43

verified ·

1 Parent(s): dc2f2e4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -34,8 +34,8 @@ controlnet = ControlNetModel.from_pretrained(
 ).to("cuda")
 pipe = StableDiffusionControlNetPipeline.from_pretrained(
-    "digiplay/GhostMixV1.2VAE",
-    #"CompVis/stable-diffusion-v1-4",
     controlnet = controlnet,
     torch_dtype = torch.float16,
     safety_checker = None,

 ).to("cuda")
 pipe = StableDiffusionControlNetPipeline.from_pretrained(
+    #"digiplay/GhostMixV1.2VAE",
+    "benjamin-paine/stable-diffusion-v1-5",
     controlnet = controlnet,
     torch_dtype = torch.float16,
     safety_checker = None,